프로그래머스(programmers) 더 맵게 python 정답 [힙(heap)]

프로그래머스(programmers) 더 맵게 python 정답 [힙(heap)]

문제 링크

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

프로그래머스

문제 정답

입출력 예

scoville	K	return
[1, 2, 3, 9, 10, 12]	7	2

 

정답 코드

from heapq import heappush,heapify,heappop
def solution(scoville, K):
    answer = 0    
    q = scoville[:]
    heapify(q)
    
    while q:
        first = heappop(q)
        if first >=K:
            return answer
        if not q:
            return -1
        second =  heappop(q)
        newItem = first + second * 2
        heappush(q,newItem)
        answer +=1       
        
    return -1

문제 해설

우선순위 큐 === 힙이다.

first >=K 부분에서 첫번째 아이템을 검사하고 조건에 해당하면 answer를 리턴한다.

만약 q가 비어져 있으면, 불가능한 조건이므로 -1을 리턴한다.

그 다음 부분은 문제의 설명에 따라 섞는 것을 단순히 구현하는 부분이다.