프로그래머스 더 맵게 (Level : 2)

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

코딩테스트 연습 - 더 맵게

 

힙을 사용하기 위해 파이썬의 heapq를 import 해야 합니다.

heap을 사용할 때 새로 list를 선언한 후 하나씩 데이터를 넣는 방법과

heapify를 이용해 주어진 list를 한 번에 힙으로 만드는 방법이 있습니다.

해당 문제에서는 heapify를 이용해서 문제를 풀어야만 제대로 된 성능과 결괏값을 출력하므로 heapify 사용법을 익힐 필요가 있습니다.

 

 

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