프로그래머스 실패율

실패율

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

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

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

전체 스테이지의 개수 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) {
    	/**
         * 구조체를 담은 PriorityQueue 이용
         */
        PriorityQueue<rateOfFail> queue = new PriorityQueue<>(new Comparator<rateOfFail>() {
            @Override
            public int compare(rateOfFail o1, rateOfFail o2) {
            	// value 값 기준 내림차순
                if(o1.value < o2.value) return 1;
                else if(o1.value > o2.value) return -1;
                else if(o1.value == o2.value) {
                	// value 값 같을 경우, idx 기준 오름차순
                    if(o1.idx > o2.idx) return 1;
                }
                return 0;
            }
        });

        int[] answer = new int[N];

		// 아직 못 깬 user count
        int[] notCleared = new int[N+2];
        // 도달한 user Count
        int[] cleared = new int[N+2];

        for(int i=0; i<stages.length; i++) {
            notCleared[stages[i]]++;
            for(int j=1; j<=stages[i]; j++) {
                cleared[j]++;
            }
        }

        double[] result = new double[N];
        for(int i=0; i<result.length; i++) {
        	// 실패율 계산해서 PriorityQueue에 담기
            if(notCleared[i+1] == 0) result[i] = 0;
            else result[i] = notCleared[i+1] * 1.0 / cleared[i+1];
            queue.offer(new rateOfFail(i, result[i]));
        }

        int idx = 0;
        while(!queue.isEmpty()) {
        	// 정렬된 PriorityQueue에서 idx 뽑기 (stage 1단계부터 시작이므로 +1)
            answer[idx++] = queue.poll().idx+1;
        }

        return answer;
    }

    static class rateOfFail {
    	// idx, value(실패율 - rateOfFail)
        public int idx;
        public double value;

        public rateOfFail(int idx, double value) {
            this.idx = idx;
            this.value = value;
        }
    }
    
}

 

URL: https://school.programmers.co.kr/learn/courses/30/lessons/42889

프로그래머스