본문 바로가기

Algorithm/프로그래머스

[프로그래머스] Lv.2 더 맵게

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

 

정답코드

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를 적용해보았기 때문에 문제없이 풀 수 있었다.