본문 바로가기
프로그래머스/1단계

(Python/LV1) 명예의 전당(1)

by windy7271 2022. 11. 27.
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에서 가장 작은 항목을 팝하고 반환합니다

반응형

댓글