[프로그래머스 / Java] 실패율

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

프로그래머스

---

◎ 문제

---

◎ 코드 및 풀이

 

import java.util.*;
class Solution {
    public int[] solution(int N, int[] stages) {
        int[] answer = {};
        
        Map<Integer,Double> hm = new HashMap<>();
        // 각 스테이지에 머물러 있는 사람들의 수
        int[] stayStagePeople = new int[N+2];
    
        for(int stage : stages){
            stayStagePeople[stage]++;
        }
        
        // 각 스테이지를 클리어한 사람들의 수
        int[] clearStagePeople = new int[N+2];
        
        for(int stage : stages){
            for(int i=0; i<stage; i++){
                clearStagePeople[i+1]++;
            }
        }
    
        
        // 0으로 나누거나 0을 나눌 경우에 NaN처리되기 때문에 처리해줘야 한다.
        for(int i =1; i<=N; i++){
            if(stayStagePeople[i]!=0&&clearStagePeople[i]!=0){
                hm.put(i,(double)stayStagePeople[i]/clearStagePeople[i]);
            } else {
                hm.put(i,0.0);
            }
            
        }
        
        
        //실패율 값으로 스테이지 번호를 내림차순 정렬
        List<Integer> list = new ArrayList<>(hm.keySet());
        
        Collections.sort(list,(o1,o2) -> Double.compare(hm.get(o2), hm.get(o1)));
        answer = list.stream().mapToInt(Integer::intValue).toArray();
        
        return answer;
    }
}

---

◎ 기록

처음에 각 실패율을 구하는 것 까지 구했는데 마지막에 스테이지 번호를 내림차순으로 정렬하는 것을 못해서 시간을 엄청 허비했다.... 

두 가지 방법이 있었는데

1. Comparator 구현체를 이용해 커스터마이징해서 정렬하는 방법

2. 람다식을 이용하는 방법

나는 람다식을 이용했다.

 

그 후 새롭게 알게 된 사실은 여태 Integer(참조객체)로 담긴 list를 배열로 옮길때 항상 for문으로 하나하나 옮겨줬었는데 stream()의 mapToInt(Integer::intValue)로 toArray() 해주면 코드 한 줄로 간단하게 int[] 로 바꿀 수 있다는 것이다.

Collection과 stream은 코테에서 필수다. 공부하자