[알고리즘] 프로그래머스 - 피로도

문제

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

프로그래머스

 

풀이

완전 간단한 구현 문제다. 일반적으로 파이썬 내장 조합이나 순열 사용하면 시간초과가 터지는 경우가 많은데 이건 그냥 쉬운 문제여서 그런가... 

탐험할 수 있는 모든 던전 탐험 순열을 찾아서 하나하나 탐험해보고 가장 많이 탐험할 수 있는 경우를 출력했다.

 

코드

from itertools import permutations

def solution(k, dungeons):
    dun_num = len(dungeons)
    answer = 0
    
    for permut in permutations(dungeons, dun_num):
        hp = k
        count = 0
        for pm in permut:
            if hp >= pm[0]:
                hp -= pm[1]
                count += 1
        if count > answer:
            answer = count
    
    return answer