Programmers, 실패율 : Java

🔗 문제 링크

실패율

 

🔎 문제 풀이

쉬워 보이면서도 은근히 푸는데 오래 걸렸다 ,,

 

HashMap을 사용해 key값은 스테이지 번호, value값은 실패율을 저장한 후 정렬하면 되겠다는 생각을 했다.

HashMap 정렬에 대해 구글링 해보니 map의 keySet을 이용해 정렬이 가능한 것을 알게 되었다.

[Java] Map을 Key, Value로 정렬하기

 

각 스테이지마다 실패율을 구하기 위해서 주어진 배열 stages를 참고하여 배열 두 개를 만들었다. 

 

1) 스테이지에 도달했으나 아직 클리어하지 못한 플레이어의 수를 담은 배열

2) 스테이지에 도달한 플레이어 수를 담은 배열

 

배열의 인덱스가 스테이지 번호라고 생각하고, 각 스테이지 번호와 일치하는 인덱스에 수를 저장한다.

두 배열의 같은 인덱스에 들어있는 원소 값을 나누어 실패율을 계산하고, HashMap에 저장하면 된다.

만약 스테이지에 도달한 유저가 없는 경우, 실패율을 0으로 정의하는 예외 처리도 해주어야 한다.

 

Key Point

🔑 배열을 활용해 실패율을 구한다.

🔑 HashMap을 사용해 스테이지 번호(key)와 실패율(value)을 저장한다.

🔑 스테이지에 도달한 유저가 없는 경우(실패율 = 0)를 고려한다.

🔑 실패율(value)을 기준으로 내림차순으로 정렬한 후, 스테이지 번호(key)를 배열 answer에 넣어준다.

 

Java 코드

import java.util.*;

class Solution {
    public int[] solution(int N, int[] stages) {
        int[] answer = new int[N];
        
        // 스테이지에 도달했으나 아직 클리어하지 못한 플레이어 수 정보를 담은 배열
        
        int[] stageCount = new int[N+2];
        
        for (int stage : stages) {
            stageCount[stage]++;
        }
        
        // 스테이지에 도달한 플레이어 수 정보를 담은 배열
        
        int[] stagePassCount = new int[N+1];
        
        stagePassCount[N] = stageCount[N] + stageCount[N+1];
        
        for (int i=N-1; i>=1; i--) {
            stagePassCount[i] = stageCount[i] + stagePassCount[i+1];
        }
        
        // HashMap을 사용해 실패율 저장 (Key: 스테이지 번호, value: 실패율)
        
        HashMap<Integer,Double> map = new HashMap<>();
        
        for (int i=1; i<=N; i++) {
            
            if (stagePassCount[i] == 0) {
                map.put(i, (double)0);
            }
            else {
                map.put(i, (double)stageCount[i]/stagePassCount[i]);
            }
        }
        
        // 각 스테이지 번호(key)를 실패율(value)의 내림차순으로 정렬
        
        List<Integer> keySet = new ArrayList<>(map.keySet());
        keySet.sort((o1, o2) -> map.get(o2).compareTo(map.get(o1)));
        
        int idx = 0;
        for (int stage : keySet) {
            answer[idx++] = stage;
        }
        
        return answer;
    }
}