[프로그래머스] 실패율 with JAVA

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

코딩테스트 연습 - 실패율

슈퍼 게임 개발자 오렐리는 큰 고민에 빠졌다. 그녀가 만든 프랜즈 오천성이 대성공을 거뒀지만, 요즘 신규 사용자의 수가 급감한 것이다. 원인은 신규 사용자와 기존 사용자 사이에 스테이지 차이가 너무 큰 것이 문제였다.

이 문제를 어떻게 할까 고민 한 그녀는 동적으로 게임 시간을 늘려서 난이도를 조절하기로 했다. 역시 슈퍼 개발자라 대부분의 로직은 쉽게 구현했지만, 실패율을 구하는 부분에서 위기에 빠지고 말았다. 오렐리를 위해 실패율을 구하는 코드를 완성하라.

• 실패율은 다음과 같이 정의한다.
  • 스테이지에 도달했으나 아직 클리어하지 못한 플레이어의 수 / 스테이지에 도달한 플레이어 수

전체 스테이지의 개수 N, 게임을 이용하는 사용자가 현재 멈춰있는 스테이지의 번호가 담긴 배열 stages가 매개변수로 주어질 때, 실패율이 높은 스테이지부터 내림차순으로 스테이지의 번호가 담겨있는 배열을 return 하도록 solution 함수를 완성하라.

제한사항

• 스테이지의 개수 N은 1 이상 500 이하의 자연수이다.
• stages의 길이는 1 이상 200,000 이하이다.
• stages에는 1 이상 N + 1 이하의 자연수가 담겨있다.
  • 각 자연수는 사용자가 현재 도전 중인 스테이지의 번호를 나타낸다.
  • 단, N + 1 은 마지막 스테이지(N 번째 스테이지) 까지 클리어 한 사용자를 나타낸다.
• 만약 실패율이 같은 스테이지가 있다면 작은 번호의 스테이지가 먼저 오도록 하면 된다.
• 스테이지에 도달한 유저가 없는 경우 해당 스테이지의 실패율은 0 으로 정의한다.

import java.util.*;

class Solution {
    static int[] counter;
    public int[] solution(int N, int[] stages) {
        counter = new int[N+2];
        Stage[] stage_list = new Stage[N];
        count(stages);
        for (int i = 1; i <= N; i++){
            float fail = calc(i, N);
            stage_list[i-1] = new Stage(i, fail);
        }
        Arrays.sort(stage_list);

        int[] ans = new int[N];
        for (int i = 0; i < stage_list.length; i++){
            ans[i] = stage_list[i].index;
        }
        return ans;
    }
    public void count(int[] stages){
        for (int index : stages) {
            counter[index]++;
        }
    }
    public float calc(int index, int N){
        int sum = 0;
        for (int i = index; i <= N+1; i++){
            sum += counter[i];
        }
        if (sum == 0) return 0;
        else return (float)counter[index] / (float)sum;
    }
    public class Stage implements Comparable<Stage> {
        int index;
        float fail;

        public Stage(int index, float fail){
            this.index = index;
            this.fail = fail;
        }

        @Override
        public int compareTo(Stage o1){
            if (this.fail == o1.fail) return this.index - o1.index;
            else if (this.fail - o1.fail < 0) return 1;
            else return -1;
        }
    }
}

1. 어려웠던 점:

- 정렬 조건에 번호는 오름차순이고, 실패율은 내림차순인 것을 착각했다. 정렬 조건을 제대로 보자.

2. 알게된 점:

- 배열의 정렬 복습

3. 알고리즘 풀이:

- count함수로 counter 배열에 각 스테이지에 도달해있는 인원을 담아준다.

calc 함수는 실패율을 계산하는 함수이다.

counter 배열을 이용해, stage 별로 실패율과 인덱스를 저장해놓은 Stage 클래스의 배열(stage_list)을 만든다.

이 후, 이 stage_list를 정렬 조건에 맞게 정렬해준 후, 인덱스값만 ans 배열에 차례대로 담으면 답이 된다.