[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 = new int[N];
        HashMap<Integer, Double> map = new HashMap<>();

        Arrays.sort(stages);

        for(int i=1; i<N+1; i++){

            double mo = 0;
            double ja = 0;

            for(int j=0; j<stages.length; j++){
                if(stages[j]>=i){
                    mo++;
                }
                if(stages[j]==i){
                    ja++;
                }
            }

            double tt;

            if(mo == 0){
                tt = 0.0;
            }else{
                tt = ja/mo;
            }

            if(tt == Double.NaN){
                tt = 0.0;
            }

            map.put(i, tt);
        }

        List<Integer> keySet = new ArrayList<>();

        keySet.addAll(map.keySet());

        keySet.sort(new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return map.get(o2).compareTo(map.get(o1));
            }
        });

        for(int i=0; i<answer.length; i++){
            answer[i] = keySet.get(i);
        }

        return answer;
    }
}

 

0.0 / 0.0 을 하면 값이 0.0이 되는게 아니라 NaN이 된다. 그래서 꼭 NaN체크를 해야한다.