[Python] Programmers 더 맵게

1. 문제 링크

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

코딩테스트 연습 - 더 맵게

 

2. 문제 요약

가장 맵지 않은 음식의 스코빌 지수 + (두 번째로 맵지 않은 음식의 스코빌 지수 * 2) 구하여 K 이상 음식 만들기

 

3. 아이디어 정리

• heapq를 이용하여 최소 힙 구한다.

 

4. 문제 풀이

4-1. 내 풀이

import heapq

def solution(scoville, K):
    answer = 0
    heapq.heapify(scoville)
    while scoville[0] < K:
        try:
            first = heapq.heappop(scoville)
            second = heapq.heappop(scoville)
            heapq.heappush(scoville, first + (second * 2))
        except IndexError:
            return -1
        answer += 1
    return answer

 

 

5. 결론

• 우선순위 큐 이용하기