programmers - 더 맵게

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

코딩테스트 연습 - 더 맵게

• 이 문제는 heap을 이용한 문제이다.
• 처음에 while 조건문을 num<K 로 둬서 계속 백점이 나오지않았다.
• heap[0]<K로 수정해서 정답을 맞췄다.
• 2021.05.19 - [Algorithm/알고리즘 및 자료구조] - 자료구조 우선순위 큐(Priority Queue)와 힙(heap)

자료구조 우선순위 큐(Priority Queue)와 힙(heap)

위에 내가 정리한 글이다. Heap 구조만 알면 풀 수 있는 문제였다.

import heapq

def solution(scoville, K):
    heap = []
    answer = 0
    for i in scoville:
        heapq.heappush(heap,i)
    num = 0
    while heap[0]<K:
        if len(heap)==1:
            return -1
        a = heapq.heappop(heap)
        b = heapq.heappop(heap)
        num = a+b*2
        heapq.heappush(heap,num)
        answer+=1
        if heap[0]>K:
            return answer