[프로그래머스] 실패율 / Python

크게 어려운 부분은 없는 문제였습니다.
정렬 후 이진탐색을 통해 쉽게 풀 수 있는 문제입니다.

programmers.co.kr/learn/courses/30/lessons/42889#

코딩테스트 연습 - 실패율

from bisect import bisect_left, bisect_right

def solution(N, stages):
    answer = []
    P = len(stages) # 사람 수
    stages.sort()
    for i in range(1, N+1):
        idx1 = bisect_left(stages, i)
        idx2 = bisect_left(stages, i+1)
        # P - idx1 해당 스테이지이상 도달
        # P - idx2 해당 스테이지를 넘어감
        if P - idx1 <= 0:
            answer.append((0, i))
        else:
            answer.append(( (idx2 - idx1) / (P - idx1), i))
    answer = sorted(answer, key = lambda x : (-x[0], x[1]))
    answer = [ ans[1] for ans in answer]
    return answer