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

문제 링크: https://school.programmers.co.kr/learn/courses/30/lessons/42626

프로그래머스

코드

import heapq

def solution(scoville, K):
    cnt = 0
    heapq.heapify(scoville)
    while scoville[0] < K:
        new = heapq.heappop(scoville) + heapq.heappop(scoville)*2
        heapq.heappush(scoville,new)
        cnt += 1
        if len(scoville) == 1 and scoville[0] < K:
            return -1
    return cnt

풀이

• 힙을 사용하는 문제이다.
• 기존 리스트를 힙으로 만들기 위해서는 heapq.heapify(scoville)를 사용하면 된다.(디폴트는 최소힙)
• 힙에서 pop을 하지않고 최솟값만 확인하려면 0번째 인덱스를 확인하면 된다.
• 종료 조건은 힙에 있는 원소의 개수가 1개이면서 그 원소가 K보다 작을때이다.(홀수이면 더이상 섞을수 없기 때문)