최소힙(가장 작은값은 언제나 인덱스 0에 저장) 사용을 위해 heapq 모듈을 사용했다
heapify를 통해 리스트를 힙으로 변형하여 쉽게 문제를 해결했다
소스 코드
import heapq
def solution(scoville, K):
heapq.heapify(scoville)
count = 0
while scoville[0] < K:
if len(scoville) > 1:
heapq.heappush(scoville, heapq.heappop(scoville) + (heapq.heappop(scoville)*2))
count += 1
else:
return -1
return count