[Swift 알고리즘] 프로그래머스 LV1 - 34. 실패율 (2019 KAKAO BLIND RECRUITMENT)

Updated:

프로그래머스 코딩테스트 연습문제 LV1 - 34. 실패율 (2019 KAKAO BLIND RECRUITMENT)

34. 실패율 (2019 KAKAO BLIND RECRUITMENT)

문제 설명

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

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

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

전체 스테이지의 개수 N, 게임을 이용하는 사용자가 현재 멈춰있는 스테이지의 번호가 담긴 배열 stages가 매개변수로 주어질 때, 실패율이 높은 스테이지부터 내림차순으로 스테이지의 번호가 담겨있는 배열을 return 하도록 solution 함수를 완성하라.킨 후 터트려져 사라진 인형의 개수를 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]

주어진 코드

import Foundation

func solution(_ board:[[Int]], _ moves:[Int]) -> Int {
    return 0
}

제출한 코드

처음 제출한 코드

  • 테스트 케이스 5, 9, 22번 시간초과로 통과 못함
  • Swift의 sort 함수 속도 문제. 다른 방법 사용해야함
// 실패율 계산
func calcFailRate(_ notClearPlayers: Int, _ stagePlayers: Int) -> Double {
    guard notClearPlayers != 0 else {
        return 0
    }
    guard stagePlayers != 0 else {
        return 1
    }
    return Double(notClearPlayers) / Double(stagePlayers)
}

// 지정한 스테이지까지 도달했으나 아직 클리어하지 못한 사용자의 수 카운팅
func countNotClearPlayers(_ stageNum: Int, _ stages: [Int]) -> Int {
    var count = 0

    for s in stages {
        if s == stageNum {
            count += 1
        }
    }

    return count
}

// 지정한 스테이지까지 도달한 사용자의 수 카운팅
func countStagePlayers(_ stageNum: Int, _ stages: [Int]) -> Int {
    var count = 0

    for s in stages {
        if s >= stageNum {
            count += 1
        }
    }

    return count
}

func solution(_ N:Int, _ stages:[Int]) -> [Int] {
    var stageList = [Int:Double]()

    for i in 1...N {
        let stagePlayers = countStagePlayers(i, stages)
        let notClearPlayers = countNotClearPlayers(i, stages)
        let failRate = calcFailRate(notClearPlayers, stagePlayers)

        stageList[i] = failRate
    }

    // 키 값으로 먼저 오름차순 정렬후 밸류 값으로 내림차순 정렬
    let sorted = stageList.sorted(by: <).sorted(by: {$0.value > $1.value})

    return sorted.map { $0.key }
}

다른 사람의 풀이 참고 후 수정한 코드

func solution(_ N:Int, _ stages:[Int]) -> [Int] {
    var fail:Dictionary = [Int:Double]()

    // 실패율 계산 후 저장
    for i in 1...N{
        // 도달한 수
        let a = stages.filter{$0 >= i}.count

        // 클리어한 수
        let b = a - stages.filter{$0 > i}.count

        // 실패율
        let failCount = Double(b) / Double(a)

        fail[i] = failCount
    }

    // 실패율이 높은 순서대로 내림차순 숫자 등록, 실패율이 같으면 오름차순
    let failSorted = fail.sorted(by: <).sorted(by: {$0.value > $1.value})

    return failSorted.map{$0.key}
}

Leave a comment