[프로그래머스] 더 맵게 - 파이썬

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

코딩테스트 연습 - 더 맵게

import heapq

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

https://www.daleseo.com/python-heapq/

[파이썬] heapq 모듈 사용법

이 문제를 풀기위해서는 heapq(힙) 이라는 모듈을 사용해야한다.

힙은 최단경로 특히 다익스트라 알고리즘에 많이 사용되므로 꼭 알아두자!

문제조건을 조금 대충? 읽어서 생각보다 오래 걸렸다

(while문 안에 else: 조건을 안걸어서 틀림)

 

이 문제의 핵심 아이디어는 k보다 작은 애들만 집중적으로 공략하여 값의 크기를 올려주면 끝이다.