파이썬 알고리즘 : 실패율

카카오 코딩테스트

2024년 4월 12일 알고리즘 문제풀이

문제

난이도

Lv. 1

코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
def solution(N, stages):
    answer = []
    arr = [0 for _ in range(N+2)]
    arr_f = [0 for _ in range(N+2)]
    for i in range(len(stages)):
        now = stages[i]
        for j in range(1,now+1):
            arr[j] += 1
        arr_f[now] += 1
    tmp = []
    for i in range(1,N+1):
        if arr[i] == 0:
            tmp.append([0,i])
        else:
            tmp.append([(arr_f[i]/arr[i]),i])
    tmp.sort(reverse=True, key=lambda x : (x[0],-x[1]))
    for i in range(N):
        answer.append(tmp[i][1])
    return answer