실패율 - 프로그래머스(Swift)

실패율 - 프로그래머스(Swift)

 

 

 문제 설명

 

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

프로그래머스

 

 

 

 

 나의 풀이

 

import Foundation

func solution(_ N:Int, _ stages:[Int]) -> [Int] {
    
    var totalUser = [Int](repeating: 0, count: N+2)
    var stopUser = [Int](repeating: 0, count: N+2)
    var answerArr = [(Int, Float)](repeating: (0, 0.0), count: N+1) 
    stages.forEach { (stage) in
        for i in (1...stage) {
            totalUser[i] += 1
        } 
        stopUser[stage] += 1
    }
    
    for i in (1...N) {
        answerArr[i] = (i, Float(stopUser[i]) / Float(totalUser[i]))
    }
    
    print(answerArr)
    
    return answerArr.sorted { $0.1 > $1.1 }.map {$0.0}.filter{$0 != 0}
    
}

 

고차함수를 이용해서 풀려고 하다가 시간이 좀 걸렸다.. 💧

실패율처럼 비율을 나타내는 것들은 실수형으로 해야한다는 것도 항상 염두해둬야겠다.

 

 

 

 

 다른 사람의 풀이

 

굉장히 직관적이고 논리적이여서 가져왔다.

import Foundation

func solution(_ N:Int, _ stages:[Int]) -> [Int] {
 
    var failure: [Int: Float] = [:]
    var player: Int = stages.count
    
    for i in 1...N {
        let n = stages.filter { $0 == i }.count
        failure[i] = Float(n)/Float(player)
        player -= n
    }

    return failure.sorted(by: <).sorted(by: { $0.value > $1.value }).map {$0.key}
}

 

1부터 N까지 돌면서 탐색하는 i가 stages에서 멈춰있는 스테이지 숫자와 같을 때를 n으로 지정하면 이 n은 특정 숫자까지 도달했지만 깨지못한 경우가 된다.

그리고 player에서는 전체 숫자에서 탐색할때마다 n을 빼줌으로써 그 다음 탐색하는 값은 이전에 값들이 도달할 수 없는 것을 표현했다.

예시로 [2, 3, 4, 5]라는 stages배열이 있을때 i가 1부터 5까지 돌면서 2가 되었다고 가정해보자. 이때 2인 값이 하나가 있어서 player.count인 4에서 1을 빼준다. 그리고 다음 i가 3이될때는 player.count가 3인 상태로 시작된다. 그 이유가 2는 3에 도달하지 못한 숫자기 때문이다. 

 

하지만 이 코드는 시간 초과가 난다. 

stages의 길이가 200,000이고 n은 500까지기 떄문에 이 둘을 곱해보면 1억이 된다. 

그래서 직관적이고 좋은 코드지만 시간복잡도 측면에서는 아쉬운 코드인 것 같다. (코딩테스트에서는)

 

 

 피드백

 

이 밖에도 여러 코드를 살펴보았는데 직관적이지만 시간초과가 나오는 코드가 굉장히 많았다.

매 순간 코드의 간결함과 성능 사이에서 적절함을 찾는 것이 중요한 것 같다.