[프로그래머스] 실패율

✨️ 문제 설명

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

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

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

전체 스테이지의 개수 N, 게임을 이용하는 사용자가 현재 멈춰있는 스테이지의 번호가 담긴 배열 stages가 매개변수로 주어질 때, 실패율이 높은 스테이지부터 내림차순으로 스테이지의 번호가 담겨있는 배열을 return 하도록 solution 함수를 완성하라.

제한사항

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

 -> 자세한 내용 보러가기

🎲 자바 풀이

import java.util.*;
class Solution {
    public int[] solution(int N, int[] stages) {
        int[] answer = new int[N];
        //스테이지별로 실패율을 저장할 배열
        double[] failStages = new double[N + 2];
        
        //각 스테이지 별로 실패한 사람을 넣음
        for(int stage : stages){
            failStages[stage]++;
        }
        //최초 스테이지에서 도전한 사람은 stages의 길이
        int totalPeople = stages.length;
        
        for(int idx = 1; idx <= N+1; idx++){
            if(totalPeople == 0){
                failStages[idx] = 0;
            }else{
                int fail = (int) failStages[idx];
                failStages[idx] = failStages[idx] / totalPeople;
                totalPeople -= fail;
            }
        }
        
        //스테이지를 저장할 리스트
        List<Integer> list = new ArrayList<>();
        for(int idx = 1; idx <=N; idx++){
            list.add(idx);
        }
        
        //리스트를 실패율을 기준으로 정렬한다
        Collections.sort(list, (a, b) -> {
            if (failStages[b] == failStages[a]) {
                return Integer.compare(a, b);
            } else {
                return Double.compare(failStages[b], failStages[a]);
            }
        });
        
        for(int idx = 0; idx < N; idx++){
            answer[idx] = list.get(idx);
        }
        
        return answer;
    }
}

풀이 설명

: 각 스테이지 별로 실패율을 구해야하므로, 스테이지별로 실패율을 저장할 배열을 선언하고, 거기에 실패한 사람의 수를 넣는다.
: 처음 시작할 때는 stages의 길이 만큼 사람이 있고, 반복문을 통해 실패율을 구한다.
: 임시적으로 스테이지를 저장할 List를 선언하고, 리스트의 순서를 실패율에 따라 순서를 바꾸고(Collections.sort), 이를 배열로 반환한다.