[프로그래머스] 더 맵게

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

프로그래머스

 

import heapq
def solution(scoville, K):
    answer = 0
    heapq.heapify(scoville)

    while scoville[0] < K:
        if len(scoville) < 2:
            return -1
        a = heapq.heappop(scoville)
        b = heapq.heappop(scoville)
        scoville_scale = a + (b * 2)
        heapq.heappush(scoville, scoville_scale)
        answer += 1


    return answer

알게된 점

• heapify 사용하면 기존의 리스트를 힙 자료형으로 바꿀 수 있다.
• heappop()은 가장 작은 원소를 제거하는 동시에 그를 결괏값으로 리턴한다.
• heap에서 가장 작은 원소를 삭제하지 않고 가져오고 싶으면 heap[0]을 사용하면 된다.