[프로그래머스] 더 맵게

출처: 프로그래머스 코딩 테스트 연습, https://programmers.co.kr/learn/challenges

 

 

 

import heapq

def solution(scoville, K):
    if K == 0: return K
    answer = 0
    heapq.heapify(scoville)
    
    while scoville[0] <= K:
        if len(scoville) == 1:
            return -1
        first = heapq.heappop(scoville)
        second = heapq.heappop(scoville)
        heapq.heappush(scoville, first + second * 2)

        answer += 1
        
    return answer