[python] 프로그래머스_실패율 (정렬)

문제

 

풀이

우선 스테이지 번호와 실패율을 함께 저장할 딕셔너리 answer을 만든다. 스테이지를 정렬하고 스테이지에서 for문을 돌면서 각 인덱스를 카운트해서 0개이면 answer에 0을, 그렇지 않으면 answer에 실패율을 계산해서 넣고 유저수는 카운트했던 수는 빼준다. 

 

for문을 다 돌고 answer을 실패율에 따라 정렬해서 리턴해준다.

def solution(N, stages):
    answer = {}
    stages.sort()
    numUser = len(stages)

    for i in range(1,N+1):
        cnt = stages.count(i)
        if cnt==0:
            answer[i]=0
        else:
            answer[i]=(cnt/numUser)
            numUser=numUser-cnt

    answer = dict(sorted(answer.items(),key=lambda x:x[1],reverse=True))
    answer = list(answer.keys())
    return answer