[Heap] 더 맵게 - 프로그래머스

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

코딩테스트 연습 - 더 맵게

 

heapq 연습 문제로 적합한듯

 

import heapq as hp
def solution(scoville, K):
    answer = 0
    # heap화 시킴
    hp.heapify(scoville) ## 최소 힙으로 사용
    ## K 이상으로 만들 수 있을 때
    while(scoville[0] < K and len(scoville)>=2) : 
        ## 2개 뽑음
        mins = hp.heappop(scoville)
        sub_mins =  hp.heappop(scoville)
        ## 새 거 만듦
        new = mins+sub_mins*2
        hp.heappush(scoville, new)
        answer += 1
    ## 만들 수 없을 때
    if scoville[len(scoville)-1] < K : 
        answer = -1
    ## 나머지는 이미 다 K 이상일 때
    return answer