프로그래머스 87946 - 피로도

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

코딩테스트 연습 - 12주차

문제 설명

XX게임에는 피로도 시스템(0 이상의 정수로 표현합니다)이 있으며, 일정 피로도를 사용해서 던전을 탐험할 수 있습니다. 이때, 각 던전마다 탐험을 시작하기 위해 필요한 "최소 필요 피로도"와 던전 탐험을 마쳤을 때 소모되는 "소모 피로도"가 있습니다. "최소 필요 피로도"는 해당 던전을 탐험하기 위해 가지고 있어야 하는 최소한의 피로도를 나타내며, "소모 피로도"는 던전을 탐험한 후 소모되는 피로도를 나타냅니다. 예를 들어 "최소 필요 피로도"가 80, "소모 피로도"가 20인 던전을 탐험하기 위해서는 유저의 현재 남은 피로도는 80 이상 이어야 하며, 던전을 탐험한 후에는 피로도 20이 소모됩니다.

이 게임에는 하루에 한 번씩 탐험할 수 있는 던전이 여러개 있는데, 한 유저가 오늘 이 던전들을 최대한 많이 탐험하려 합니다. 유저의 현재 피로도 k와 각 던전별 "최소 필요 피로도", "소모 피로도"가 담긴 2차원 배열 dungeons 가 매개변수로 주어질 때, 유저가 탐험할수 있는 최대 던전 수를 return 하도록 solution 함수를 완성해주세요.

 

입출력 예

k	dungeons	result
80	[[80,20],[50,40],[30,10]]	3

---

문제 풀이

첫번째로 시도해본 방법은 dungeons 배열을 정렬해서 [i][0]번째 인덱스가 큰 것부터 차례차례 대로 하면 되겠지라고 생각했지만 입출력 예를 만족할 수 없었고, 그 반대로 정렬해도 만족하지 않았다.

이문제는 애초에 정렬로 풀 수 있는 문제가 아니었다.

그냥 모든 경우의 수를 생각하는 문제였다.

k도 5000까지이고 던전의 수도 8개라서 그렇게까지 효율성을 생각하지 않아도 되는 문제라서 모든 경우의 수를 생각하기로 하였다.

bfs, dfs를 사용해도 되지만 난 그냥 모든 경우의 수를 permutation으로 뽑고 계산하기로 하였다.

from itertools import permutations

def solution(k, dungeons):
    answer = 0
    n = len(dungeons)

    sample = [i for i in range(n)]
    permute = list(permutations(sample, n))

    for data in permute:
        tmp = k
        temp = 0
        for i in data:
            if tmp >= dungeons[i][0]:
                tmp -= dungeons[i][1]
                temp += 1
        if temp > answer:
            answer = temp

    return answer