문제 설명
-
실패율
= 스테이지에 도달했으나 아직 클리어하지 못한 플레이어의 수 / 스테이지에 도달한 플레이어 수 -
전체 스테이지의 개수 N, 게임을 이용하는 사용자가 현재 멈춰있는 스테이지의 번호가 담긴 배열 stages가 매개변수로 주어질 때, 실패율이 높은 스테이지부터 내림차순으로 스테이지의 번호가 담겨있는 배열을 return 하도록 solution 함수를 완성하라.
- 입출력 예 설명
#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]내가 작성한 코드
- 시간 초과로 실패한 코드
func solution(_ N: Int, _ stages: [Int]) -> [Int] {
var pass: [Int] = []//스테이지에 도달한 사람의 수
var fail: [Int] = [] //도달했지만 스테이지를 실패한 사람의 수
var doubleRate: [Double] = [] //단계별 실패율을 요소로 갖는 배열
var rate: [Int] = []
for i in 0..< N {
pass.append(stages.filter{$0 >= (i + 1)}.count)
fail.append(stages.filter{$0 == (i + 1)}.count)
if pass[i] != 0 {
doubleRate.append(Double(fail[i]) / Double(pass[i])) //각 스테이지의 실패율
} else {
doubleRate.append(0.0)
}
}
for i in 0..<N {
//rate에 실패율이 높은 순서대로 (인덱스 + 1)을 넣어줘서 리턴해야함
rate.append(doubleRate.firstIndex(of: doubleRate.sorted{$0 > $1}[i])! + 1)
doubleRate[doubleRate.firstIndex(of: doubleRate.sorted{$0 > $1}[i])!] = 2
}
return rate
}5, 9, 22번에서 시간초과가 발생하여 통과하지 못했다. stages의 길이는 최대 200,000이다. filter를 통해 stages에서 조건을 만족하도록 필터링해주는 것도 굉장한 시간이 들 것이다.
나는 통과한 사람의 수와, 실패한 사람의 수를 구해서 실패율을 구하려고 했다. 하지만 다시 생각해보면 해당 스테이지에 도달한 사람으로 실패한 사람의 수를 구할 수 있었다.
즉, 1단계에 도달한 사람이 8명이고 2단계에 도달한 사람이 7명이라면 1단계에서 실패한 사람은 8 - 7인 한 명이 되는 것이다.
- 다른 분 코드를 참고하여 작성한 코드
func solution(_ N: Int, _ stages: [Int]) -> [Int] {
var rate: [Int: Double] = [:]
var fail = [Int](repeating: 0, count: N + 1)
for i in stages {
for j in 0..<i {
fail[j] += 1
}
}
for i in 0..<N {
rate.updateValue(Double(fail[i] - fail[i+1]) / Double(fail[i]), forKey: i + 1)
}
return rate.sorted {$0.key < $1.key}.sorted{$0.value > $1.value}.map{$0.key}
}마지막 단계를 통과한 사람은 N + 1의 값을 갖기 때문에(5단계를 통과하면 stages의 요소에는 6이 들어가있음) fail도 N + 1의 칸을 만들어 주었다.
for문을 이용해 각 스테이지에 도달한 사람이 얼마나 되는지 fail이라는 배열에 추가해주었다. stages의 요소가 2일 경우엔 그 사람이 1단계를 통과하고 2단계까지 도달했다는 뜻이므로 , fail의 [0]과 [1]에 1씩 더해주어 1단계, 2단계에 도달한 사람을 한 명 추가해주었다.
다음으로 다시 for문을 이용하여 위에서 말한 식을 이용해 실패율을 구해주고 Dictionary를 통해 i+1의 키값에 각각 도달한 사람의 수를 넣어주었다. 원래 0번째 인덱스가 1단계를 의미하고 있었기 때문에 i + 1에 넣어주었다.
같은 실패율을 가질 경우에 작은 숫자가 앞에 오도록 sorted로 정렬을 먼저 한 뒤, 다시 sorted를 이용해서 value에 대해 정렬한 뒤 map을 이용해 키 값으로 이루어진 배열을 리턴해주었다.
개인적으로 시간 초과를 해결하다 힘들어서 다른 분들의 코드를 참고하였다. 지금이 시작일테니 내 스스로 효율적인 코드를 짜도록 노력해야겠다.
