[알고리즘] 프로그래머스 더 맵게 / python

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

코딩테스트 연습 - 더 맵게 | 프로그래머스

 

정렬하고 최소값을 찾을때는 최소힙으로 풀어야 시간복잡도 낮출 수 있다

import heapq

def solution(scoville, K):
    answer = 0
    heapq.heapify(scoville)
    
    while True:
        min1 = heapq.heappop(scoville)
        if min1 >= K:
            break
        elif len(scoville) == 0:
            answer = -1
            break
        min2 = heapq.heappop(scoville)
        _sum = min1 + 2*min2
        heapq.heappush(scoville, _sum)
        answer += 1
    
    return answer