프로그래머스 - 더 맵게

알고리즘 문제풀이

문제 링크

# 1. 런타임 에러 + 효율성 시간초과
# 2. heapq로 변경하였음에도 런타임 에러
# 3. try ~ except 로 수정
import heapq

def solution(scoville, K):
    heap = []
    for num in scoville:
        heapq.heappush(heap, num)
    answer = -1
    cnt = 0
    while heap[0] < K:
        try:
     		heapq.heappush(heap, heapq.heappop(heap) + (heapq.heappop(heap) * 2))
        except IndexError:
        	return -1
        cnt += 1
    return cnt

참고 소스