프로그래머스 : 더 맵게

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

코딩테스트 연습 - 더 맵게

 

해결 방법 : 최소 heap

import heapq

def solution(scoville, K):
    answer = 0
    heapq.heapify(scoville) # 힙 구성
    
    while len(scoville) > 1: # 하나 남을때까지
        food1 = heapq.heappop(scoville)
        food2 = heapq.heappop(scoville)
        if food1 >= K:
            return answer
        
        newFood = food1 + (food2 * 2) # 음식 조합
        heapq.heappush(scoville, newFood)
        answer += 1
    
    return answer if scoville[0] >= K else -1