[Programmers] [2019 KAKAO BLIND RECRUITMENT] [Python] Level1_실패율

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

코딩테스트 연습 - 실패율

풀이

처음 내가 푼 풀이이다. 딕셔너리를 이용하여 풀려고 했지만 시간복잡도가 높아서 인지 70.4 점을 받게 되었다. 

# 프로그래머스 1단계 - 실패율
from collections import defaultdict

n = int(input()) # 실패율
stages = list(map(float, input().split())) # stages 

def solution(n, stages):
    answer = []
    dictionary = defaultdict(float)
    res_dict = defaultdict(float)
    a = len(stages)
    for i in range(1, n+1):
        dictionary[i] = stages.count(i)
        res_dict[i] = stages.count(i)
    for i in range(1, n+1):
        res_dict[i] = dictionary[i] / a 
        a = a - dictionary[i]
    new_dict = sorted(res_dict.items(),reverse=True, key = lambda x: x[1])   
    for key in new_dict:
        answer.append(key[0])    
    return answer
print(solution(n, stages))

 

맞는 풀이

def solution(N, stages):
    answer = []
    score = []
    R = len(stages)
    for i in range(1, N + 1):
        if R != 0:
            count = stages.count(i)
            score.append((i, count / R))
            R -= count
        else:
            score.append((i, 0))

    score = sorted(score, key=lambda x: (-x[1], x[0]))
    for i in range(len(score)):
        answer.append(score[i][0])
    return answer