[프로그래머스] 실패율 (카카오, 정렬)

https://programmers.co.kr/learn/courses/30/lessons/42889?language=python3#

코딩테스트 연습 - 실패율

def solution(N, stages):
    arr = [[] for _ in range(N + 1)]
    stages.sort(reverse=True)
    stage = N + 1
    count = stages.count(N + 1)
    index = count
    for i in range(N):
        stage -= 1
        while 0 <= index < len(stages) and stages[index] == stage:
            index += 1
        if index >0:
            arr[stage] = [(index - count) / index, stage]
        else:
            arr[stage] = [0, stage]
        count = index
    arr = sorted(arr[1:], key=lambda x: (-x[0], x[1]))
    answer = [arr[i][1] for i in range(N)]
 
    return answer

정렬해서 큰 스테이지부터 숫자를 세어가면서 풀면 된다.

다만 주의할 점은, 예를 들어 N = 4, stages = [2]와 같은 값이 나오면 0으로 나누는 경우가 있을 수 있으니, 예외 처리를 해야 한다.