728x90
반응형
문제 출처: https://school.programmers.co.kr/learn/courses/30/lessons/138477
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
풀이:
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에서 가장 작은 항목을 팝하고 반환합니다
반응형
댓글