[프로그래머스] 더 맵게 - 파이썬(Python)

[프로그래머스] 더 맵게 - 파이썬(Python)

 

 

 

import heapq

def solution(scoville, K):
    heapq.heapify(scoville)
    count = 0
    while len(scoville) >= 2 and scoville[0] < K:
        first, second = heapq.heappop(scoville), heapq.heappop(scoville)
        heapq.heappush(scoville, first + (second * 2))
        count += 1
    return count if scoville[0] >= K else -1

혹은

import heapq

def solution(scoville, K):
    heapq.heapify(scoville)
    count = 0
    while scoville[0] < K:
        try:
            first, second = heapq.heappop(scoville), heapq.heappop(scoville)
            heapq.heappush(scoville, first + (second * 2))
            count += 1
        except IndexError:
            return -1
    return count

 

 

 

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

코딩테스트 연습 - 더 맵게