[programmers] 실패율 - Java

실패율

문제 설명

• 실패율은 다음과 같이 정의한다.
• 스테이지에 도달했으나 아직 클리어하지 못한 플레이어의 수 / 스테이지에 도달한 플레이어 수

제한사항

• 스테이지의 개수 N은 1 이상 500 이하의 자연수이다.

• stages의 길이는 1 이상 200,000 이하이다.

• stages에는 1 이상 N + 1 이하의 자연수가 담겨있다.
• 각 자연수는 사용자가 현재 도전 중인 스테이지의 번호를 나타낸다.
• 단, N + 1 은 마지막 스테이지(N 번째 스테이지) 까지 클리어 한 사용자를 나타낸다.

• 만약 실패율이 같은 스테이지가 있다면 작은 번호의 스테이지가 먼저 오도록 하면 된다.

• 스테이지에 도달한 유저가 없는 경우 해당 스테이지의 실패율은 0 으로 정의한다.

입출력 예

입출력 예 설명

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

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

• 3 번 스테이지 실패율 : 2/4

• 4번 스테이지 실패율 : 1/2

• 5번 스테이지 실패율 : 0/1

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

• [4,1,2,3]

solution.java

import java.util.*;

class Solution {
    public int[] solution(int N, int[] stages) {
        int[] arr = new int[N+1];
        double reachedPlayer = stages.length;
        
        for(int i=0; i<stages.length; i++){
            arr[stages[i]-1]++;
        }
        
        HashMap<Integer, Double> map = new HashMap<>();
        
        for(int i=0; i<arr.length-1; i++){
            double num = 0.0;
            if(arr[i]!=0){
                num = arr[i] / reachedPlayer;
            }
            
            map.put(i+1, num);
            reachedPlayer -= arr[i];
        }
        
        List<Integer> keySet = new ArrayList<>(map.keySet());
        int[] answer = new int[N];
        
        keySet.sort((o1, o2) -> map.get(o2).compareTo(map.get(o1)));
     
        for(int i=0; i<keySet.size(); i++){
            answer[i] = keySet.get(i);
        }
        return answer;
    }
}

핵심 키워드

• 배열을 선언해서 해당하는 사용자가 현재 멈춰있는 스테이지를 배열의 인덱스로 지정해서 값을 증가시킨다.

• HashMap을 선언해서 실패율과 인덱스를 쌍으로 저장한다.
• 각 스테이지마다 도전한 사용자는 이전 스테이지에서 클리어하지 못한 사용자를 빼면 구할 수 있다.

• HashMap의 keySet을 List에 넣은 후, List의 .sort((o1, o2) -> HashMap.get(o2).compareTo(HashMap.get(o1)) 메소드로 값을 정렬할 수 있다.

결론!