[프로그래머스] 실패율

문제

실패율을 구하는 코드를 완성하라.

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

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

제한사항

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

풀이

• [stage] : 각 스테이지에 도달한 사용자의 수
• [passNo] : 해당 스테이지를 통과한 사용자의 수
• [answer] : (스테이지 번호, 실패율)

코드

def solution(N, stages):
    answer = []
    stage = [0] * (N+1)
    for player in stages:
        stage[player-1] += 1
        
    passNo = [0] * (N+1)
    for s in range(1, N+1):
        passNo[N-s] = stage[N-s+1] + passNo[N-s+1]
        
    for i in range(N):
        answer.append((stage[i]/(passNo[i]+stage[i]) if passNo[i]+stage[i] != 0 else 0, i))
    answer.sort(key=lambda x: (-x[0],x[1]))
    return [x[1]+1 for x in answer]