[프로그래머스 | 파이썬 python] 더 맵게 | heapq를 이용한 풀이

🌏문제

프로그래머스

 

👽풀이

이거 그냥 리스트로 풀면 런타임 에러난다. heapq를 사용해야한다!

• heapq.heapify(arr) : 리스트를 heap으로 변환해줌
• heapq.heappop(arr) : heap에서 가장 작은 수 리턴
• heapq.heappush(arr, ele) : heap에 삽입

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