[프로그래머스] 실패율.java

riyu_·2021년 6월 28일
0

프로그래머스

목록 보기
7/13

문제 설명

슈퍼 게임 개발자 오렐리는 큰 고민에 빠졌다. 그녀가 만든 프랜즈 오천성이 대성공을 거뒀지만, 요즘 신규 사용자의 수가 급감한 것이다. 원인은 신규 사용자와 기존 사용자 사이에 스테이지 차이가 너무 큰 것이 문제였다.

이 문제를 어떻게 할까 고민 한 그녀는 동적으로 게임 시간을 늘려서 난이도를 조절하기로 했다. 역시 슈퍼 개발자라 대부분의 로직은 쉽게 구현했지만, 실패율을 구하는 부분에서 위기에 빠지고 말았다. 오렐리를 위해 실패율을 구하는 코드를 완성하라.

  • 실패율은 다음과 같이 정의한다.
    - 스테이지에 도달했으나 아직 클리어하지 못한 플레이어의 수 / 스테이지에 도달한 플레이어 수
    전체 스테이지의 개수 N, 게임을 이용하는 사용자가 현재 멈춰있는 스테이지의 번호가 담긴 배열 stages가 매개변수로 주어질 때, 실패율이 높은 스테이지부터 내림차순으로 스테이지의 번호가 담겨있는 배열을 return 하도록 solution 함수를 완성하라.

제한사항

  • 스테이지의 개수 N은 1 이상 500 이하의 자연수이다.
  • stages의 길이는 1 이상 200,000 이하이다.
  • stages에는 1 이상 N + 1 이하의 자연수가 담겨있다.
    - 각 자연수는 사용자가 현재 도전 중인 스테이지의 번호를 나타낸다.
    - 단, N + 1 은 마지막 스테이지(N 번째 스테이지) 까지 클리어 한 사용자를 나타낸다.
  • 만약 실패율이 같은 스테이지가 있다면 작은 번호의 스테이지가 먼저 오도록 하면 된다.
  • 스테이지에 도달한 유저가 없는 경우 해당 스테이지의 실패율은 0 으로 정의한다.

입출력 예

Nstagesresult
5[2, 1, 2, 6, 2, 4, 3, 3][3,4,2,1,5]
4[4,4,4,4,4][4,1,2,3]

입출력 예 설명

입출력 예 #1
1번 스테이지에는 총 8명의 사용자가 도전했으며, 이 중 1명의 사용자가 아직 클리어하지 못했다. 따라서 1번 스테이지의 실패율은 다음과 같다.

- 1 번 스테이지 실패율 : 1/8

2번 스테이지에는 총 7명의 사용자가 도전했으며, 이 중 3명의 사용자가 아직 클리어하지 못했다. 따라서 2번 스테이지의 실패율은 다음과 같다.

- 2 번 스테이지 실패율 : 3/7

마찬가지로 나머지 스테이지의 실패율은 다음과 같다.

- 3 번 스테이지 실패율 : 2/4
- 4번 스테이지 실패율 : 1/2
- 5번 스테이지 실패율 : 0/1

각 스테이지의 번호를 실패율의 내림차순으로 정렬하면 다음과 같다.

- [3,4,2,1,5]

입출력 예 #2

모든 사용자가 마지막 스테이지에 있으므로 4번 스테이지의 실패율은 1이며 나머지 스테이지의 실패율은 0이다.

- [4,1,2,3]

솔루션

실패율은 [실패해서 머물러있는 사람 수/현재 도달해있거나 지나간 사람 수]이다. 입출력 예 #1의 스테이지3을 따져보자면 현재 3인 사람은 2명, 현재 3이거나 3 이상인 사람은 4명으로 실패율은 2/4가 된다.

처음에는 HashMap으로 할까 했는데 이는 사람 수를 세고 저장하는 게 비효율적이었다. 그래서 단순하게 배열을 이용했다. 스테이지 이름을 HashMap에 key로 저장하는 대신 배열의 인덱스 i+1로 이름에 접근했다. 실패 사람 수는 stages 배열을 돌면서 나타나는 스테이지에 +1을 해주었다. 이 때 stages에는 N보다 큰 수가 있을 수 있는데 이건 지나치게 했다. 도달 사람 수는 stages 배열을 돌면서 그 요소보다 작거나 같은 스테이지에 +1을 했다.
이렇게 실패율을 구할 분자 배열, 분모 배열을 완성했고 최종 실패율 내림차순 배열을 만들어야 했다. 이게 은근 복병이었다. 두 개의 배열을 같은 인덱스끼리 연산하는 것은 문제없었으나 그 결과값을 내림차순으로 정렬한 후의 스테이지의 이름 즉 인덱스를 가져오는 게 어려웠다.
배열로 해결하고 싶었지만 방법이 없었고 HashMap의 key에 스테이지 이름, value에 실패율을 저장해서 value를 기준으로 정렬하고 최종 key 배열을 가져오도록 했다.

import java.util.*;
class Solution {
    public int[] solution(int N, int[] stages) {
        int[] answer = new int[N];
        
        int[] fail = new int[N];
        int[] arrival = new int[N];
        for(int i = 0; i < stages.length; i++){
            for(int j=0; j<N; j++){
                if(j+1 <= stages[i]) arrival[j]++; 
            }            

            if(stages[i] > N) continue;
            fail[stages[i]-1]++;
            
        }
        
        HashMap<Integer, Double> map = new HashMap<>();
        
        for(int i=0; i<N; i++){
            if(fail[i] == 0 | arrival[i] == 0) {
                map.put(i+1, (double)0); continue;
            }
            map.put(i+1, fail[i] / (double)arrival[i]); 
        }
        
        List<Integer> keySetList = new ArrayList<>(map.keySet());		
        
	Collections.sort(keySetList, (o1, o2) -> (map.get(o2).compareTo(map.get(o1))));
        
        for(int i=0; i<keySetList.size(); i++){
            answer[i] = keySetList.get(i);
        }
        
        return answer;
    }
}

0개의 댓글

Powered by GraphCDN, the GraphQL CDN