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

문제풀이

1. 각 스테이지에 도전 중인 사람의 수를 구합니다.

2. ing[i]를 스테이지 i에 도전 중인 사람의 수라 하면, 스테이지 i에 도달한 사람의 수는 리스트 ing에서 i번째 이상의 요소에 도전 중인 사람의 합, 즉, sum(ing[i:])가 됩니다.

예) 총 6 단계라 하면, 5단계에 도달한 사람의 수는 5단계에 도전중인 사람 수+6단계에 도전중인 사람 수+7단계에 도전중인 사람 수(마지막 단계인 6단계를 클리어한 사람 수)입니다.

3. 스테이지 i의 실패율은 (i단계에 도전중인 사람 수)/(스테이지 i 이상에 도전중인 사람 수)가 됩니다. (ing[i]/sum(ing[i:]))

Python code

#https://programmers.co.kr/learn/courses/30/lessons/42889#
from collections import Counter
def solution(N, stages):
    tot=len(stages)
    c=Counter(stages)
    
    # stage i에 도전중인 사람 수
    ing = [c[i] if i in c else 0 for i in range(N+2)]
    #실패율
    f_rate = {}
    for i in range(N+1) :
        # i 단계의 실패율 : i 단계에 도전중인 사람 수 / i 단계 이상에 도전 중인 사람 수
        f_rate[i] = ing[i]/sum(ing[i:]) if sum(ing[i:])>0 else 0
    return sorted([i for i in range(1,N+1)], key=lambda x : f_rate[x], reverse=True)

문제 링크

코딩테스트 연습 - 실패율