[프로그래머스-Lv2] 더 맵게 / Python

코딩테스트 연습 - 더 맵게

import sys
import heapq

input = sys.stdin.readline

scoville = list(map(int, input().split()))
k = int(input())


def solution(scoville, k):
    scoville.sort()
    answer = 0
    while len(scoville) > 0:
        if scoville[0] >= k:
            return answer
        temp1 = heapq.heappop(scoville)
        if scoville:
            heapq.heappush(scoville, temp1 + (heapq.heappop(scoville) * 2))
            answer += 1
    return -1


print(solution(scoville, k))

해설

heap을 이용해 풀 수 있었던 문제다. 

별다른 정렬없이 제일 작은 값을 추출할 수 있으므로, 효율성테스트에서도 문제가 없다.