더 맵게[python3]

https://programmers.co.kr/learn/courses/30/lessons/42626?language=python3 

코딩테스트 연습 - 더 맵게

 오늘 heap을 활용해 더 맵게라는 문제를 풀었다. 

import heapq

def solution(scoville, K):
    answer = 0
    heapq.heapify(scoville)
    
    while len(scoville) >=2:   
        min1 = heapq.heappop(scoville)
        min2 = heapq.heappop(scoville)
        temp = min1 + min2*2
        heapq.heappush(scoville,temp)
        answer +=1
        if K <= scoville[0]:
            return answer      
    return -1

모든 음식의 스코빌 지수가 K 이상이어야 문제가 해결 되는 것인데

최초로 섞은 음식값이 K 값을 넘었을 때 값을 리턴해버리는 실수를 해서 해결하는데 오래걸렸다.