[프로그래머스/2단계/파이썬(Python3)] 더 맵게

문제 출처

코딩테스트 연습 - 더 맵게

 

풀이 코드

import heapq

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

 

힙의 개념을 이용하면 빠르게 풀 수 있는 문제입니다.

 

힙 개념 확인하러 가기↓↓↓↓

https://data-analysis-expertise.tistory.com/101

[자료구조/파이썬(Python3)] 힙 (Heap)