[프로그래머스] 더 맵게 (Python)

문제 링크

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

프로그래머스

 

문제 설명

모든 음식의 스코빌 지수를 K 이상으로 만들고 싶다. 모든 음식의 스코빌 지수를 K 이상으로 만들기 위해 Leo는 스코빌 지수가 가장 낮은 두 개의 음식을 아래와 같이 특별한 방법으로 새로운 음식을 만든다.

 

섞음 음식의 스코빌 지수 = 가장 맵지 않은 음식의 스코빌 지수 + (두 번째로 맵지 않은 음식의 스코빌 지수 * 2)

 

모든 음식의 스코빌 지수를 K 이상으로 만들기 위해 섞어야 하는 최소 횟수를 return 하는 함수를 작성

제한사항

• scoville의 길이는 2 이상 1,000,000 이하입니다.
• K는 0 이상 1,000,000,000 이하입니다.
• scoville의 원소는 각각 0 이상 1,000,000 이하입니다.
• 모든 음식의 스코빌 지수를 K 이상으로 만들 수 없는 경우에는 -1 을 return 합니다.

 

풀이

스코빌 리스트를 힙 자료구조로 변환하여, 힙의 인덱스 0번 째가 최소값이므로 조건을 주고,

스코빌 리스트의 길이가 1이하일 때는 -1 return 하도록 하였다.

 

✅ Code - 성공

import heapq

def solution(scoville, K):
    answer = 0
    heapq.heapify(scoville)
    
    while scoville[0] < K:
        if len(scoville) <= 1:
            return -1
        answer += 1
        sco1 = heapq.heappop(scoville)
        sco2 = heapq.heappop(scoville)
        shake_sco = sco1 + (sco2 * 2)
        heapq.heappush(scoville, shake_sco)
    
    return answer