[Swift] 프로그래머스 실패율(2019 KAKAO Blind Recruitment)(lv. 1)

[Swift] 프로그래머스 실패율(2019 KAKAO Blind Recruitment)(lv. 1)

 

1. 문제

https://school.programmers.co.kr/learn/courses/30/lessons/42889

프로그래머스

 

2. 접근

실패율의 정의를 충분하게 읽으면 된다.

"스테이지에 도달했으나 아직 클리어하지 못한 플레이어의 수 / 스테이지에 도달한 플레이어의 수"

 

주어진 배열 stages는 현재 사용자가 멈춰있는 스테이지의 번호가 담겨있다. 그렇다면, 각 스테이지 별로 몇 명의 사람이 멈춰있는지를 세고, 순차적으로 실패율을 구해나가면 답을 구할 수 있을 것이라 생각했다.

결과적으로 구해야 하는 것은 '스테이지의 실패율'을 sorting 한 것이다. 그렇기에 인덱싱에 신경을 쓰면서 배열을 활용해야 한다.

 

아래의 코드에서 이와 같은 생각을 구현해 보았다.

 

먼저 첫 번째 반복문에서는 그 스테이지에 있는 플레이어의 수를 세서 playerCount 배열에 넣어준다.

그 다음 반복에서는 playerCount를 돌면서 실패율을 구하도록 했다. stagePlayer는 현재 스테이지까지 통과한 사람의 수를 나타낸 것이다. playerCount의 원소(이 스테이지에 멈춰있는 사람의 수)를 stagePlayer로 나누고 그 원소를 stagePlayer에 빼주면, 누적적으로 현재 스테이지에 도달한 플레이어 수를 빼면서 실패율을 구할 수 있다.

 

그 다음 answerBeforeSort의 맨 마지막 수는 모든 스테이지를 통과한 사람이기 때문에 실패율 구하는 것에 쓰이지 않는다. 그렇기에 하나만 빼주고 while문으로 들어간다.

 

최대값을 찾아주고, 그 최대값이 있는 index를 찾아준다. 그러고 answer 배열에 index + 1을 하여 append 해준다. 실제 주어진 값은 인덱스 값이 아닌 '몇 번째' 스테이지라고 주어졌기 때문에 +1 을 해주어야 한다. 그 다음 answerBeforeSort에서 max값으로 다시 등장하지 않게 하기 위해 원소를 -1로 만들어준다.

 

반복문을 다 돌면 답이 구해져있다. 그것을 리턴한다.

 

3. 코드

import Foundation

func solution(_ N:Int, _ stages:[Int]) -> [Int] {
    var playerCount = [Int](repeating: 0, count: N + 1)
    for i in stages {
        playerCount[i - 1] += 1
    }
    var stagePlayer = stages.count
    var answerBeforeSort = [Double]()
    for i in playerCount {
        answerBeforeSort.append(stagePlayer != 0 ? Double(i) / Double(stagePlayer) : 0)
        stagePlayer -= i
    }
    answerBeforeSort.removeLast()
    var answer = [Int]()
    while answer.count != N {
        let max = answerBeforeSort.max()!
        let index = answerBeforeSort.firstIndex(of: max)!
        answer.append(index + 1)
        answerBeforeSort[index] = -1
    }
    return answer
}