[Level 1 / Python] 실패율(2019 KAKAO BLIND RECRUITMENT)

문제 설명

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

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

• 실패율은 다음과 같이 정의한다.

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

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

• 스테이지의 개수 N은 1 이상 500 이하의 자연수이다.

• stages의 길이는 1 이상 200,000 이하이다.

• stages에는 1 이상 N + 1 이하의 자연수가 담겨있다.

  • 각 자연수는 사용자가 현재 도전 중인 스테이지의 번호를 나타낸다.

  • 단, N + 1 은 마지막 스테이지(N 번째 스테이지) 까지 클리어 한 사용자를 나타낸다.

• 만약 실패율이 같은 스테이지가 있다면 작은 번호의 스테이지가 먼저 오도록 하면 된다.

• 스테이지에 도달한 유저가 없는 경우 해당 스테이지의 실패율은 0 으로 정의한다.

출처 : programmers.co.kr/learn/courses/30/lessons/42889

풀이 방법

예를 들어 [2, 1, 2, 6, 2, 4, 3, 3]의 값을 가지는 stages 배열의 각 원소들은 사용자별 실패한 스테이지와 같으므로 두번째 원소인 1이 의미하는 것은 이 사용자는 1번 스테이지에서 실패했다고 설명할 수 있다. 그러므로 이 문제는 먼저 입력 받은 스테이지의 개수 N에 대해 1번 스테이지부터 N번 스테이지까지 stages 배열의 같은 값을 가진 원소들을 파악하면 된다.

• 1번 스테이지에서 값 1을 가진 원소의 개수, 2번 스테이지에서 값 2를 가진 원소의 개수, ... , N번 스테이지에서 값 N을 가진 원소의 개수들을 구한다.

• 실패한 사용자는 stages 배열에서 빠지게 되므로 배열의 크기를 줄여나간다.

• stages 배열 원소의 개수가 0일 경우는 실패했다고 볼 수 있으므로 실패율은 0이라고 볼 수 있다.

위의 방법들을 가지고 스테이지별 실패율을 구해보면 다음과 같다.

stages = [2, 1, 2, 6, 2, 4, 3, 3]
answer = []
for i in range(1, 6):
    if stages:
        print("사용자   : ", stages)
        answer.append(stages.count(i) / len(stages))
        print("스테이지 : ", i)
        print("실패율   : ", answer)
        print("------------------------------------------")
        stages = list(filter(lambda x: x != i, stages))
    else:
        answer.append(0)
 
"""
사용자   :  [2, 1, 2, 6, 2, 4, 3, 3]
스테이지 :  1
실패율   :  [0.125]
------------------------------------------
사용자   :  [2, 2, 6, 2, 4, 3, 3]
스테이지 :  2
실패율   :  [0.125, 0.42857142857142855]
------------------------------------------
사용자   :  [6, 4, 3, 3]
스테이지 :  3
실패율   :  [0.125, 0.42857142857142855, 0.5]
------------------------------------------
사용자   :  [6, 4]
스테이지 :  4
실패율   :  [0.125, 0.42857142857142855, 0.5, 0.5]
------------------------------------------
사용자   :  [6]
스테이지 :  5
실패율   :  [0.125, 0.42857142857142855, 0.5, 0.5, 0.0]
------------------------------------------
"""

위 코드를 살펴 보면 stages.count(스테이지 번호)를 통해 각 스테이지에 해당하는 원소의 개수를 구하고, filter(lambda(x : x != i, stages))를 이용해 계산된 스테이지의 원소는 제외하여 stages 배열을 변환했다.마지막으로 이 문제에서는 각 스테이지의 번호를 실패율의 내림차순으로 정렬하라고 하였으므로 결과값을 key는 스테이지 번호, value는 실패율을 가지는 딕셔너리를 이용하여 실패율 기준 정렬하고 결과를 반환하도록 했다.

소스 코드

def solution(N, stages):
    answer = {}
    for i in range(1, N+1):
        if stages:
            answer[i] = stages.count(i) / len(stages)
            stages = list(filter(lambda x: x != i, stages))
        else:
            answer[i] = 0
    return sorted(answer, key=lambda x: answer[x], reverse=True)