[프로그래머스] 42626번 더 맵게

우선순위 큐 문제

 

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

 

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

 

heappop은 가장 작은 원소를 찾아서 반환후 삭제된다.