https://school.programmers.co.kr/learn/courses/30/lessons/42626
정답코드
def solution(scoville, K):
import heapq
heap = [x for x in scoville]
heap.sort()
answer = 0
while heap[0] < K and len(heap) != 1:
num = heapq.heappop(heap) + (heapq.heappop(heap) * 2)
heapq.heappush(heap, num)
answer += 1
return -1 if heap[0] < K else answer
이미 다익스트라에서 heapq를 적용해보았기 때문에 문제없이 풀 수 있었다.
'Algorithm > 프로그래머스' 카테고리의 다른 글
[프로그래머스] Lv.2 스킬트리 (0) | 2022.09.17 |
---|---|
[프로그래머스] Lv.2 오픈채팅방 (카카오) (1) | 2022.09.17 |
[프로그래머스] Lv.2 프린터 (0) | 2022.09.14 |
[프로그래머스] Lv.2 기능개발 (0) | 2022.09.14 |
[프로그래머스] Lv.2 n^2 배열 자르기 (0) | 2022.09.13 |