프로그래머스 level 2 - 더 맵게 (힙 Heap)

programmers.co.kr/learn/courses/30/lessons/42626

코딩테스트 연습 - 더 맵게

import heapq
# 최소 힙 자료구조

h = [1, 2, 3]
heapq.heapify(h)
print(heapq.heappop(h)) # 1 - 최소값 반환
heapq.heappush(h, 0)    # 입력 후 가장 작은 값이 앞에 오게 자동 정렬
print(heapq.heappop(h)) # 0 - 최소값 반환

 

import heapq

def solution(scoville, K):
    cnt = 0
    h = scoville
    heapq.heapify(h)

    while True:
        tmp = heapq.heappop(h)
        if tmp >= K:
            break
            
        if len(h) == 0:
            return -1
        tmp += (heapq.heappop(h)) * 2
        heapq.heappush(h, tmp)
        cnt += 1
    return cnt