더 맵게 [파이썬]

Moveheon·2023년 10월 13일

프로그래머스

목록 보기
10/24
import heapq

def solution(scoville, K):
    answer = 0
    heapq.heapify(scoville)
    while(True):
        if heapq.heappop(scoville) < K:
            answer += 1
            tmp1 = heapq.heappop(scoville)
            tmp2 = heapq.heappop(scoville)
            heapq.heappush(scoville, tmp1 + tmp2 * 2)
        else:
            break
    return answer + 1

0개의 댓글