명예의 전당(1)

이정연·2023년 1월 4일
0

CodingTest

목록 보기
104/165

문제 링크

풀이

heap 자료 구조를 이용하면 쉽게 풀 수 있다.

python은 최소 힙이므로 k개를 유지하며 0번 인덱스를 뽑아주면 최하 점수를 알 수 있다.

코드

import heapq as h

def solution(k, score):
    answer = []
    honor = []
    for s in score:
        h.heappush(honor,s)
        if len(honor) > k:
            h.heappop(honor)
        answer.append(honor[0])
    return answer
profile
0x68656C6C6F21

0개의 댓글