[python]실패율

2019 KAKAO BLIND RECRUITMENT>실패율

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

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

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

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

[제한 조건]

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

[입출력 예]

[입출력 예 설명]

입출력 예 #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]

나의 풀이

생각해보기!

def solution(N, stages):

    fail = {} # 실패율을 추가할 딕셔너리
    all = len(stages) # 스테이지에 도달한 플레이어 수(전체)
    
    for i in range(1,N+1):
        if all == 0: # 스테이지에 도달한 플레이어 수가 0명일 때
            fail[i] = 0
        else:
            fail[i] = stages.count(i)/(all)
            all -= stages.count(i) # 스테이지에 도달한 플레이어 수
        
    fail= sorted(fail, key=fail.get, reverse=True) 

    return fail

• 각 스테이지의 실패율을 내림차순으로 정리하기 위한 방법으로 딕셔너리가 떠올랐다. 딕셔너리 생성은 어렵지 않았지만 value값 기준 내림차순 정렬 후, key값을 얻는 코드는 검색을 통해 활용했다.
  • sorted()에 reverse = True 라는 파라미터로 내림차순을 정렬을 했다.
  • reverse 파라미터를 쓰지않으면 오름차순으로 정렬된다.
• value 기준 정렬 & key값 출력
  • dict = sorted(dict, key=dict.get)

• 실패율의 분모가 0이 되는 즉, 스테이지에 도달한 플레이어 수가 0명일 때를 고려하지 않으면 이와 같이 런타임 에러로 실패하는 테스트 케이스가 생긴다.
  • if문으로 분모가 0일 때와 아닐 때를 나누어 코드를 짰다.

런타임 에러에 대한 해결책을 질문하기에서 얻을 수 있었다. 분모가 0이 되는 경우와 같이 런타임에 영향을 끼칠 수 있는 요소에 대해 알아두자!🤔

다른 사람의 풀이 #1

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

참고하세요!

• 딕셔너리 value값에 따른 내림차순 정렬 후 key값만 출력할 때
  key=lambda x : result[x] = key=fail.get 둘 다 알아두자!

다른 사람의 풀이 #2

def solution(N, stages):
    fail = {}
    for i in range(1,N+1):
        try:
            fail_ = len([a for a in stages if a==i])/len([a for a in stages if a>=i])
        except:
            fail_ = 0
        fail[i]=fail_
    answer = sorted(fail, key=fail.get, reverse=True)
    return answer

참고하세요!

• 분모가 0이 되는 오류 예외 처리 기법으로 try, except문을 활용했다.
  • try 블록 수행 중 오류가 발생하면 except 블록이 수행된다.
  • try 블록에서 오류가 발생하지 않는다면 except 블록은 수행되지 않는다.
• 리스트 내부에 for문을 이용하여 조건에 맞는 플레이어 수를 구했다.