[프로그래머스] 더 맵게 - 파이썬/힙큐

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

import heapq

def solution(scoville, K):

    heapq.heapify(scoville)
    cnt = 0
    while len(scoville)>1:

        elem1 = heapq.heappop(scoville)

        if elem1 <K:
            elem2 = heapq.heappop(scoville)
            mix = elem1 + (elem2 *2)
            heapq.heappush(scoville,mix)
            if mix < K and len(scoville)==1:
                answer = -1
                return answer
            cnt+=1
        else:
            break

    answer = cnt

    return answer

힙큐를 활용해 조건대로 구현했다.

-1을 return하는 조건만 주의하면 된다.