BSHwan 2020. 5. 15. 13:30
반응형

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr


import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

class Solution {
    public int[] solution(int N, int[] stages) {
             int[] answer = new int[N];
        Map<Integer, Double> hash = new HashMap<Integer, Double>();
        
        double total = stages.length;
        double[] people = new double[N+2];
        double[] fail = new double[N+2];
        
        for(int i=0; i<stages.length; i++) {
        	people[stages[i]]++;
        }
        
        
        for(int i=1; i<fail.length; i++) {
        	if(total != 0) {
        		fail[i] = people[i] / total;
        	}
        	total -= people[i];
        }
        
        for(int i=1; i<fail.length-1; i++) {
//        	System.out.println(i + "번 스테이지 실패율 : " + fail[i]);
        	hash.put(i, fail[i]);
        }
        
        List<Integer> keySetList = new ArrayList<>(hash.keySet());
        
        Collections.sort(keySetList, (o1, o2) -> (hash.get(o2).compareTo(hash.get(o1))));
        
        int i = 0;
        for(Integer key : keySetList) {
//        	System.out.println(String.format("Key : %s, Value : %s", key, hash.get(key)));
        	answer[i] = key;
        	i++;
        }

        return answer;
    }
}
반응형