[프로그래머스] L1 실패율 / 2019 카카오 블라인드 채용 (Java)

카카오

[프로그래머스] L1 실패율 / 2019 카카오 블라인드 채용

[풀이]

import java.util.*;
class Solution {
    class FR implements Comparable<FR>{
        int stage;
        double rate;
        public FR(int stage, double rate){
            this.stage = stage;
            this.rate = rate;
        }
        @Override
        public int compareTo(FR f){
            if(this.rate > f.rate)
                return -1;
            else if(this.rate < f.rate)
                return 1;
            else{
                if(this.stage > f.stage)
                    return 1;
                else if(this.stage < f.stage)
                    return -1;
                return 0;
            }
        }
    }
    public int[] solution(int N, int[] stages) {
        int[] fail = new int[N+2]; // 모든 스테이지(N+1:전부 해결)를 고려한 배열
        for(int i=0; i<stages.length; i++) // 각 스테이지에 도달한 유저
            fail[stages[i]]++;
        
        int fin = fail[N+1]; // 전부 해결한 경우
        List<FR> list = new ArrayList<FR>();
        for(int i=N;  i>=1; i--){
            // 계산해서 list에 추가
            double bunja = fail[i];
            double bunmo = fail[i+1] + fail[i];
            fail[i] += fail[i+1];
            list.add(new FR(i,bunja/bunmo));
        }
        Collections.sort(list);
        int[] answer = new int[list.size()];
        for(int i=0; i<list.size(); i++){
            answer[i] = list.get(i).stage;
        }
        return answer;
    }
}