[Programmers.Lv1] 실패율

문제

전체 스테이지(N)	사용자가 현재 멈춰있는 스테이지 번호 (stages)	결과
5	[2, 1, 2, 6, 2, 4, 3, 3]	[3,4,2,1,5]

 

입출력 예 #1
1번 스테이지에는 총 8명의 사용자가 도전했으며, 이 중 1명의 사용자가 아직 클리어하지 못했다. 따라서 1번 스테이지의 실패율은 다음과 같다.

• 1 번 스테이지 실패율 : 1/8

2번 스테이지에는 총 7명의 사용자가 도전했으며, 이 중 3명의 사용자가 아직 클리어하지 못했다. 따라서 2번 스테이지의 실패율은 다음과 같다.

• 2 번 스테이지 실패율 : 3/7

마찬가지로 나머지 스테이지의 실패율은 다음과 같다.

• 3 번 스테이지 실패율 : 2/4
• 4번 스테이지 실패율 : 1/2
• 5번 스테이지 실패율 : 0/1

각 스테이지의 번호를 실패율의 내림차순으로 정렬하면 다음과 같다.

• [3,4,2,1,5]

 

풀이

스테이지의 개수(N)가 최대 500이므로 0~501 까지의 크기를 갖는 index 배열을 선언한다.

(배열에 1을 더해주는 이유는 게임 스테이지가 0이 아닌 1부터 시작하기 때문)

이 배열은 테스트 케이스로 주어지는 stages 배열 내에 존재하는 수의 갯수를 카운팅하기 위한 배열이다.

 

stages 배열이 아래와 같이 주어지는 경우, index에는 [0, 1, 3, 2, 1, 0, 1, 0, 0, 0, 0, 0,  ... ] 처럼 값이 채워지게 된다.

1번 스테이지를 클리어하지 못한 플레이어 수 = 1명, 2번 스테이지를 클리어하지 못한 플레이어수 = 3명...

이를 사용하여 실패율을 계산한다.

 

실패율은 클리어 하지 못한 플레이어 수 / 스테이지에 도달한 플레이어수 로 계산할 수 있으므로,

전체 플레이어수인 8을 기준으로 1번 스테이지부터 계산해주면 된다.

실패율 계산 후에는 1번 스테이지를 클리어하지 못한 플레이어수를 제외하고 전체 플레이어수를 다시 계산해주어야 한다.

(1번 스테이지를 클리어하지 못한 플레이어가 1명이므로, 2번 스테이지를 계산할 때는 전체 플레이어수가 7이 됨)

 

이 단계를 마치고 나면 arr 에는 0.125, 0.428, 0.5, 0.5, 0.0 이 들어가있다. 이를 내림차순으로 정렬해 주어야 한다.

이때 실패율만 정렬해주면 되는게 아니라, 실패율에 해당하는 스테이지를 최종 출력해야하므로 maxIndex 변수를 별도로 선언해주었다.

반복문을 돌면서 차례대로 가장 큰 실패율이 담겨있는 곳의 위치(스테이지)를 찾고, 이 위치를 answer 배열에 넣어준다.

(넣고 난 후에는 해당 배열의 위치에 -1을 넣어 주어야 이후 재계산하지 않음)

 

* 주의

실패율 계산 시 0으로 나누는 경우에 NaN이 입력되므로,

실패율을 계산할 arr 배열은 먼저 0으로 초기화 해놓고, 0으로 나눌 경우(플레이어 수가 0이 되었을 경우) break 해주어야 함.

 

 

코드

import java.util.Arrays;

public class 실패율 {
    public static void main(String[] args) {
        // Test case
        int N = 8; // 500 이상
        int stages[] = {1, 2, 3, 4, 5, 6, 7}; // 200000 이상

        solution(N, stages);
    }

    public static int[] solution(int N, int[] stages) {
        int[] answer = new int[N];
        int[] index = new int[500+2]; // 0 ~ 501

        // 실패율을 계산할 실수형 배열
        double[] arr = new double[N];
        Arrays.fill(arr, 0);

        // 배열 요소 카운트
        int stagesLen = stages.length;
        for(int i=0; i<stagesLen; i++) {
            index[stages[i]]++;
        }

        // 실패율 계산
        double fail = 0;
        for(int i=0; i<N; i++) {
            if(stagesLen == 0) break; // NaN

            fail = (double)index[i+1] / stagesLen;
            arr[i] = fail;

            // 사용자 수 감소
            stagesLen = stagesLen - index[i+1];
        }

        // 실패율의 내림차순 정렬
        int maxIndex = 0, answerIndex = 0;
        while(answerIndex < N) {
            for (int i = 0; i < N; i++) {
                if (arr[i] > arr[maxIndex]) {
                    maxIndex = i;
                }
            }
            answer[answerIndex] = maxIndex + 1;
            arr[maxIndex] = -1;
            answerIndex++;
        }
        
        return answer;
    }
}