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

https://programmers.co.kr/learn/courses/30/lessons/42889

코딩테스트 연습 - 실패율

import sys

input = sys.stdin.readline

n = int(input())
stages = list(map(int, input().split()))


def solution(n, stages):
    answer = {}
    length = len(stages)
    for stage in range(1, n + 1):
        if length != 0:
            cnt = stages.count(stage)
            answer[stage] = cnt / length
            length -= cnt
        else:
            answer[stage] = 0
    return sorted(answer, key=lambda x: answer[x], reverse=True)


print(solution(n, stages))

 

해설

answer를 딕셔너리로 정의해준다. 스테이지별로 실패율을 저장한다.

 

각 스테이지까지만 해결한 인원 수를 제외하면 다음 스테이지에 도달한 인원이 된다. -> length

스테이지에 도달한 인원이 없을 때(length==0)는 실패율을 0이므로 answer[stage]에 0을 저장한다.

 

실패율을 내림차순으로 정렬한 값을 리턴해준다.