[programmers] 실패율

실패율

x
슈퍼 게임 개발자 오렐리는 큰 고민에 빠졌다. 그녀가 만든 프랜즈 오천성이 대성공을 거뒀지만, 요즘 신규 사용자의 수가 급감한 것이다. 원인은 신규 사용자와 기존 사용자 사이에 스테이지 차이가 너무 큰 것이 문제였다.
​
이 문제를 어떻게 할까 고민 한 그녀는 동적으로 게임 시간을 늘려서 난이도를 조절하기로 했다. 역시 슈퍼 개발자라 대부분의 로직은 쉽게 구현했지만, 실패율을 구하는 부분에서 위기에 빠지고 말았다. 오렐리를 위해 실패율을 구하는 코드를 완성하라.
​
실패율은 다음과 같이 정의한다.
스테이지에 도달했으나 아직 클리어하지 못한 플레이어의 수 / 스테이지에 도달한 플레이어 수
전체 스테이지의 개수 N, 게임을 이용하는 사용자가 현재 멈춰있는 스테이지의 번호가 담긴 배열 stages가 매개변수로 주어질 때, 실패율이 높은 스테이지부터 내림차순으로 스테이지의 번호가 담겨있는 배열을 return 하도록 solution 함수를 완성하라.
​
제한사항
스테이지의 개수 N은 1 이상 500 이하의 자연수이다.
stages의 길이는 1 이상 200,000 이하이다.
stages에는 1 이상 N + 1 이하의 자연수가 담겨있다.
각 자연수는 사용자가 현재 도전 중인 스테이지의 번호를 나타낸다.
단, N + 1 은 마지막 스테이지(N 번째 스테이지) 까지 클리어 한 사용자를 나타낸다.
만약 실패율이 같은 스테이지가 있다면 작은 번호의 스테이지가 먼저 오도록 하면 된다.
스테이지에 도달한 유저가 없는 경우 해당 스테이지의 실패율은 0 으로 정의한다.

 

오랜만에 풀어보는 프로그래머스 문제라서 레벨 1부터 도전!

레벨 1이기도하고 그림도 귀엽고 내용도 쉬워보였는데, 막상 코드를 짜보니 ㅠㅠ 깔끔하게 짜지지가 않는다.

제출을 해보니 정확도 테스트만 하는데, 그런 줄 알았으면 마지막 for문을 map의 value값으로 sort하여서 진행해도 됐을 것 같다.

 

HashMap을 사용해서 하다가, 실패율이 같을 경우 스테이지 오름차순으로 기입을 하도록 되어있어 key값을 정렬해주는 TreeMap을 사용해서 진행했다.

 

x
import java.util.Arrays;
import java.util.Collections;
import java.util.TreeMap;
​
class Solution {
    public int[] solution(int N, int[] stages) {
        int[] answer = new int[N];
        int total = stages.length;
        
        TreeMap<Integer, Float> map = new TreeMap<>();
        
        // 스테이지 사용자 count
        for (int stage : stages) {
            if (stage > N) continue;
            
            float count = map.containsKey(stage) ? map.get(stage) + 1 : 1f;
            map.put(stage, count);
        }
        
        // 실패율 계산
        for (int i = 1; i <= N; i++) {
            float count = map.containsKey(i) ? map.get(i) : 0f;
            
            if (total == 0) map.put(i, 0f);
            else            map.put(i, (float) count/total);
            
            total -= count;
        }
        
        int p = 0;
        
        // 정렬
        for (int i = 1; i <= N; i++) {
            if (p >= N) break;
            
            if ((float) map.get(i) == Collections.max(map.values())) {
                answer[p++] = i;
                map.put(i, -1f);
                i = 0;
            }
        }
​
        return answer;
    }
}