알고리즘 공부/코딩테스트(프로그래머스)

[프로그래머스 Level 2] 피로도 (Python 파이썬) 풀이 방법

yepppi 2023. 1. 10. 17:06
반응형
SMALL

오늘은 하루종일 프로젝트 코딩을 했다!! 곧 출시 예정이라 떨린다.. 오늘은 어제 퇴근하기 전에 풀었던 문제를 리뷰하려 한다.

 

 

프로그래머스 <피로도> 문제 보기

https://school.programmers.co.kr/learn/courses/30/lessons/87946

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

 

프로그래머스 <피로도> 문제 풀이

from itertools import permutations

def solution(k, dungeons):
    answer = -1
    array = list(permutations(dungeons, len(dungeons)))
    
    for each in array:
        p = k
        check = 0
        for dungeon in each:
            if p >= dungeon[0]:
                p -= dungeon[1]
                check += 1
            else:
                continue
                
        if answer < check:
            answer = check
            
    return answer

완전탐색으로 풀어야 하는 문제였다. 던전을 탐색할 수 있는 모든 방법을 permutations로 구해 array에 리스트 형태로 담아주었다. 각 리스트의 튜플 안의 던전을 돌며 피로도를 계산해주었다. 하나의 튜플을 다 돌았을 때 탐색 가능한 던전의 개수를 answer와 비교해주어 answer에는 최댓값을 담아줄 수 있도록 했다.

 

 

 

다른 풀이

solution = lambda k, d: max([solution(k - u, d[:i] + d[i+1:]) + 1 for i, (m, u) in enumerate(d) if k >= m] or [0])

한 줄로 풀어낸 코드가 있어 가지고왔다... 아직까지도 잘 이해가 안되는데 혹시 댓글로 설명해주실 분 계신가요...

 

 

 

이 외에도 DFS나 BFS를 사용하여 완전탐색을 한 풀이도 있었다. 파이썬으로 DFS, BFS를 어떻게 잘 푸는지 공부를 좀 해야 되는데... 알고리즘은 어려워 ㅠㅠ

반응형
LIST