[Programmers][Python] Lv.2 더 맵게

Overview

• 체감 난이도: ★☆☆☆☆
• 문제 레벨: Lv.2 
• 문제 유형: 힙(heap)
• 풀이 상태: 답안참고 / 스스로 해결
• 추후: 다시 풀어보기 / 간단 복습 / 완벽 이해 

---

[문제]

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

프로그래머스

 

 

[코드]

heapq.heappop 배열 내 최솟값을 pop 해준다. 

import heapq

def solution(scoville, K):
    answer = 0
    heapq.heapify(scoville)
    while scoville[0] < K:
        if len(scoville) >= 2:
            m1 = heapq.heappop(scoville)
            m2 = heapq.heappop(scoville)

            heapq.heappush(scoville, m1 + (m2 * 2))
            answer += 1
        else:
            answer = -1
            break

    return answer