프로그래머스) 실패율 (JAVA)

문제 설명

실패율

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

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

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

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

제한사항

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

입출력 예

 

N	stages	result
5	[2, 1, 2, 6, 2, 4, 3, 3]	[3,4,2,1,5]
4	[4,4,4,4,4]	[4,1,2,3]

입출력 예 설명

입출력 예 #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]

입출력 예 #2

모든 사용자가 마지막 스테이지에 있으므로 4번 스테이지의 실패율은 1이며 나머지 스테이지의 실패율은 0이다.

• [4,1,2,3]

 

해결 과정

Arrays.sort 함수로 comparator 를 계산하였다. comparator에는 실패율의 첫 0번째 인덱스 값을 내림차순으로 정렬하는데 실패율이 같다면 그대로 두는 방식을 사용하였다.

 

또한 도달한 사람이 없다면 실패율을 -1로 만들어서 제일 뒤쪽에 정리되도록 만들었다.

 

정답 코드

import java.util.*;
class Solution {
    public int[] solution(int n, int[] stages) {
		int[] clear = new int[n+2];
		float[][] fail = new float[n][2];
		int[] answer = new int[n];
		int[] person = new int[n+2];
		
		for(int i: stages) clear[i]++;
		
		
		person[0] = stages.length;
		for(int i=1; i<n+2; i++) {
			person[i] = person[i-1] - clear[i-1];
		}
		
		for(int i=1; i<n+1; i++) {
			if(person[i] == 0) fail[i-1][0] = -1;
			else 
				fail[i-1][0] = clear[i]/(float)person[i];
			fail[i-1][1] = i;
		}
		
		Arrays.sort(fail, new Comparator<float[]>() {
			@Override
			public int compare(float[] o1, float[] o2) {
				// TODO Auto-generated method stub
				if(o1[0] == o2[0]) return 0;
				return (o1[0] < o2[0])? 1:-1;
			}
		});
		
		for(int i=0; i<n; i++) {
			answer[i] = (int) fail[i][1];
		}
		
		return answer;
    }
}

 

이클립스용 실행 코드

package Level1;

import java.util.Arrays;
import java.util.Comparator;

public class Error_Rate {
	public static void main(String[] args) {
		int N = 10;
		int[] stages = {1,2,3,4,5,6,7};
		int[] ans = solution(N, stages);
		
		System.out.println(Arrays.toString(ans));
	}

	private static int[] solution(int n, int[] stages) {
		int[] clear = new int[n+2]; // 현 스테이지에 몇 사람이 있는지
		float[][] fail = new float[n][2]; // 각 스테이지별 실패 확률[0] 스테이지[1]
		int[] answer = new int[n];
		int[] person = new int[n+2]; //각 스테이지별 통과한 사람의 수
		
		for(int i: stages) clear[i]++; // 현 스테이지에 총 몇명이 있는지를 나타냄
		
		
		person[0] = stages.length;
		for(int i=1; i<n+2; i++) { // 통과한 사람의 수 계산
			person[i] = person[i-1] - clear[i-1];
		}
		
		for(int i=1; i<n+1; i++) { //실패확률 계산
			if(person[i] == 0) fail[i-1][0] = -1; // 현 스테이지에 도달한 사람이 없다면 실패율은 -1
			else 
				fail[i-1][0] = clear[i]/(float)person[i];
			fail[i-1][1] = i;
		}
		
		Arrays.sort(fail, new Comparator<float[]>() { // 실패율을 오름차순 정리
			@Override
			public int compare(float[] o1, float[] o2) {
				// TODO Auto-generated method stub
				if(o1[0] == o2[0]) return 0;
				return (o1[0] < o2[0])? 1:-1;
			}
		});
		
		for(int i=0; i<n; i++) { //스테이지를 int화 시킴.
			answer[i] = (int) fail[i][1];
		}
		
		return answer;
	}
}