프로그래머스 [Lv. 2] 더 맵게 {언어 : Python}

문제 링크

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

프로그래머스

정답 코드

import heapq
def solution(scoville, K):
    answer = 0
    heap = []
    for s in scoville:
        heapq.heappush(heap, s)
    
    while heap and heap[0] < K:
        answer += 1
        first = heapq.heappop(heap)
        if heap:
            second = heapq.heappop(heap)
            tmp = first + second * 2
            heapq.heappush(heap, tmp)
        else:
            answer = -1
            break

    return answer

풀이 방법

1. heap에 scoville을 전부 넣는다.
   1. heapq.heapify(scoville)로 scoville을 heap으로 만드는 방법도 있지만 속도는 저게 더 빠른 것 같다.
2. heap이 존재하고 heap의 0번째 인덱스(가장 작은 값)이 K보다 작으면
   1. 두 번째로 heap이 존재하는지 확인한다.
      1. 존재하면 second를 꺼내서 tmp를 만들고 다시 넣어준다.
      2. 넣으면서 섞었다는 표시를 하기 위해 answer에 1을 더한다.
   2. 존재하지 않는다면 K이상으로 만들 수 없기 때문에 answer 을 -1로 치환하고 종료한다.
3. answer를 반환한다.

느낀점

• python은 heapq를 제공해주기 때문에 너무 편리하다.