728x90
반응형
문제 출처: https://school.programmers.co.kr/learn/courses/30/lessons/138477
풀이:
import heapq
def solution(k, score):
save, res = [], []
for i in score:
if len(save) < k:
heapq.heappush(save,i)
else:
heapq.heappushpop(save,i)
res.append(save[0])
return res
heappushpop(save, i ) 는
힙에 i 를 푸시한 다음, heap에서 가장 작은 항목을 팝하고 반환합니다
반응형
댓글