[프로그래머스] 실패율

문제

programmers.co.kr/learn/courses/30/lessons/42889

코딩테스트 연습 - 실패율

문제해결방법

스테이지의 개수만큼 배열을 선언해주고 각 스테이지마다 도전중인 사용자의 수를 저장해준다.

스테이지를 앞부터 돌며 (현재 스테이지에 도전중인 사용자의 수 / 스테이지까지 도달한 인원의 수) 로 실패율을 구해준다.

다음 스테이지로 넘어갈 때 남은 사용자의 수를 구하기 위해 도전중인 사용자의 수를 빼면서 계산해준다.

전체 스테이지에 대해 실패율을 구하고 조건에 맞춰 정렬해준다.

코드

package Programmers;
 
import java.util.Arrays;
import java.util.Comparator;
import java.util.PriorityQueue;
 
public class Solution_실패율 {
 
    public static int[] solution(int N, int[] stages) {
        int people = stages.length;
        int[] failPerStage = new int[N+2];
        PriorityQueue<double[]> pq = new PriorityQueue<>(new Comparator<double[]>() {
            @Override
            public int compare(double[] arg0, double[] arg1) {
                if(arg0[1]==arg1[1]) return Double.compare(arg0[0], arg1[0]);
                return Double.compare(arg1[1], arg0[1]);
            }
        });
        for(int stage : stages) failPerStage[stage]++;
        for(int i=1; i<=N; i++) {
            if(people!=0) pq.add(new double[]{i, (double)failPerStage[i]/people});
            else pq.add(new double[]{i, 0});
            people-=failPerStage[i];
        }
        int[] answer = new int[N];
        for(int i=0; i<N; i++) answer[i] = (int)pq.poll()[0];
        return answer;
    }
    
    public static void main(String[] args) {
        int N = 4;
        int[] stages = {4,4,4,4,4};
        System.out.println(Arrays.toString(solution(N, stages)));
    }
}

코드에 대한 피드백이나 더 좋은 아이디어는 언제나 환영입니다.