[프로그래머스]피로도

lee-goeun·2022년 5월 24일
0

문제링크
https://programmers.co.kr/learn/courses/30/lessons/87946

문제 설명

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

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

제한사항

  • k는 1 이상 5,000 이하인 자연수입니다.
  • dungeons의 세로(행) 길이(즉, 던전의 개수)는 1 이상 8 이하입니다.
    • dungeons의 가로(열) 길이는 2 입니다.
    • dungeons의 각 행은 각 던전의 ["최소 필요 피로도", "소모 피로도"] 입니다.
    • "최소 필요 피로도"는 항상 "소모 피로도"보다 크거나 같습니다.
    • "최소 필요 피로도"와 "소모 피로도"는 1 이상 1,000 이하인 자연수입니다.
    • 서로 다른 던전의 ["최소 필요 피로도", "소모 피로도"]가 서로 같을 수 있습니다.

입출력 예

문제풀이

  1. dungeons배열 요소의 두번 째 값을 더한 값이 k보다 작거나 같으면 배열 요소 차가 큰 값으로 정렬한다.
  2. k보다 크면 두번째 요소를 기준으로 내림차순 정렬한 뒤 첫번째 요소를 기준으로 오름차순 정렬한다.
  3. dungeons배열을 돌며 max값이 dungeons 첫번째 요소보다 크거나 같으면 max를 두번째 값만큼 빼주고 cnt를 1 증가한다.
  4. max보다 크면 break를 걸어 빠져나온다.
  5. cnt가 answer보다 크면 answer에 cnt를 넣어주고 dungeons길이만큼 반복해서 실행한다.

코드

function solution(k, dungeons) {
    var answer = 0;
    let cnt = 0;
    let num = 0;
    let sum = dungeons.reduce((acc, cur) => acc + cur[1], 0);
    if(sum <= k){
        dungeons.sort((a,b) => (b[0]-b[1]) - (a[0]-a[1]));
    }else{
        dungeons.sort((a,b) => {
            if(a[1] < b[1]) return -1;
            if(a[1] > b[1]) return 1;
            
            if(a[0] > b[0]) return -1;
            if(a[0] < b[0]) return 1;
        });
    }
    while(num <= dungeons.length){
       let max = k;
        for(let i=0; i<dungeons.length; i++){
            if(max >= dungeons[i][0]){
                max = max - dungeons[i][1];  
                cnt++;
            }else break;
        }
        dungeons.unshift(dungeons.pop());
        num++;
        if(cnt > answer) answer = cnt;
        cnt = 0;
    }
    
    return answer;
}

완전탐색이 안되서 테스트 통과가 안되는 것 같다. 완전탐색을 할 수 있도록 dfs를 써야 하는 거 같은데 아직 dfs가 익숙하지 않다. 좀 더 공부해보고 다시 풀어봐야겠다.

다른사람 코드

function solution(k, d) {
    const N = d.length
    const visited = new Array(N).fill(0)
    let ans = 0

    function dfs(k, cnt){
        ans = Math.max(cnt, ans)

        for (let j = 0; j < N; j++){
            if (k >= d[j][0] && !visited[j]){
                visited[j] = 1
                dfs(k - d[j][1], cnt + 1)
                visited[j] = 0
            }
        }
    }

    dfs(k, 0)
    return ans;
}

TIL

  • 방문확인을 위해 배열 생성하여 true, false로 관리한다.

0개의 댓글