[python] 프로그래머스 더 맵게 (Lv.2)

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

코딩테스트 연습 - 더 맵게

import heapq

def solution(scoville, K):
    answer = 0
    heapq.heapify(scoville)
    
    while True:        
        if scoville[0] >= K:
            break
        elif len(scoville) == 1:
            answer = -1
            break
        scoville.append(heapq.heappop(scoville)+heapq.heappop(scoville)*2)
        answer += 1

    return answer

시간초과나서 heapq 모듈 사용해서 풀었더니 성공

 

# heapq 함수

- heapq.heappush(heap, item) : item을 heap에 추가

- heapq.heappop(heap) : heap에서 가장 작은 원소를 pop & 리턴. 비어 있는 경우 IndexError가 호출됨

- heapq.heapify(x) : 리스트 x를 즉각적으로 heap으로 변환