[프로그래머스] 더 맵게(heap) 파이썬

나의 풀이

import heapq

def solution(scoville, K):
    answer = 0
    
    #heapq로 변환
    heapq.heapify(scoville)
    
    #loop 돌기
    while True:
        # 최솟값 pop
        temp_min = heapq.heappop(scoville)

        if temp_min >= K :
            return answer
        elif len(scoville) < 1:
            return -1
        else:
            temp_min2 = heapq.heappop(scoville)
            temp_min = temp_min + (temp_min2 * 2)    
            
            heapq.heappush(scoville, temp_min)
            answer += 1

heap 자료구조

참고 gmlwjd9405 github