[Python] 2019 KAKAO BLIND RECRUITMENT, 실패율

이하 문제는 https://programmers.co.kr/ 에 나와있는 문제임을 밝힙니다. 

 

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

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

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

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

제한사항

• 스테이지의 개수 N은 1 이상 500 이하의 자연수이다.
• stages의 길이는 1 이상 200,000 이하이다.
• stages에는 1 이상 N + 1 이하의 자연수가 담겨있다.
  • 각 자연수는 사용자가 현재 도전 중인 스테이지의 번호를 나타낸다.
  • 단, N + 1 은 마지막 스테이지(N 번째 스테이지) 까지 클리어 한 사용자를 나타낸다.
• 만약 실패율이 같은 스테이지가 있다면 작은 번호의 스테이지가 먼저 오도록 하면 된다.
• 스테이지에 도달한 유저가 없는 경우 해당 스테이지의 실패율은 0 으로 정의한다.

크게 어려운 문제는 아니지만, 처음 문제를 풀 당시 런타임(Runtime) 오류가 발생하였습니다. 런타임 오류가 발생하는 이유는 여러가지가 있지만 Python에서는 보통 주어진 메모리 제한을 초과할 때 발생합니다.

 

위 문제에서 런타임 오류가 발생하는 이유는 stages 변수의 길이가 200,000이기 때문에 발생할 가능성이 높습니다. 

따라서 사전에 이를 적절하게 처리해야합니다.

 

기본적으로 코드를 짜면 다음과 같습니다. 

def solution(N, stages):
    temp = dict()
    gamer = len(stages) # 참여자 수
    count = 0

    for i in range(1, N+1): 
      count = stages.count(i) # 스테이지에 머물러 있는 사람의 수
      fail = count / gamer # 실패율 계산
      temp[i] = fail # key값을 스테이지 번호로, value 값을 실패율로 가지는 딕셔너리
      gamer -= count # 전 스테이지의 사람은 전체 gamer에서 빼줌

    return sorted(temp, key=lambda x: temp[x], reverse=True) # 내림차순으로 정렬

하지만 이 코드를 그대로 실행시키면 Runtime 오류가 발생합니다. 

적절하게 처리하는 방법은, if문을 통해 마지막 조건 "스테이지에 도달한 유저가 없는 경우 해당 스테이지의 실패율은 0 으로 정의한다."  을 해결하면 됩니다. 

 

최종적으로 코드를 구성하면 다음과 같습니다.

def solution(N, stages):
    temp = dict()
    gamer = len(stages)
    count = 0

    for i in range(1, N+1):
        if gamer != 0: # 조건 추가
            count = stages.count(i)
            fail = count / gamer
            temp[i] = fail

            gamer -= count
        else:
            temp[i] = 0
    
    return sorted(temp, key=lambda x: temp[x], reverse=True)