[프로그래머스] 더 맵게 /파이썬 /Python / heapq

💡solutions )

✅ 처음에 heapq 자료구조 사용하지 않고 sort를 했다가 런타임 에러가 떴었다. 다른 분들은 처음부터 heapq를 사용하시길 바란다.

✅ 참고로 heapq는 heappush, heappop을 할 때 자동으로 정렬을 해주어 효율성이 높아진다고 한다.

 

🎫code )

import heapq
def solution(scoville, K):
    items = []
    for i in scoville:
        heapq.heappush(items, i)
    cnt = 0
    while items[0] < K:
        try:
            heapq.heappush(items, heapq.heappop(items)+(heapq.heappop(items)*2))
        except:
            return -1
        cnt += 1
    return cnt

 

📌 description )

문제출처 : programmers.co.kr/learn/courses/30/lessons/42626

코딩테스트 연습 - 더 맵게

문제 설명 매운 것을 좋아하는 Leo는 모든 음식의 스코빌 지수를 K 이상으로 만들고 싶습니다. 모든 음식의 스코빌 지수를 K 이상으로 만들기 위해 Leo는 스코빌 지수가 가장 낮은 두 개의 음식을 아래와 같이 특별한 방법으로 섞어 새로운 음식을 만듭니다. Leo는 모든 음식의 스코빌 지수가 K 이상이 될 때까지 반복하여 섞습니다. Leo가 가진 음식의 스코빌 지수를 담은 배열 scoville과 원하는 스코빌 지수 K가 주어질 때, 모든 음식의 스코빌 지수를 K 이상으로 만들기 위해 섞어야 하는 최소 횟수를 return 하도록 solution 함수를 작성해주세요. 제한 사항 scoville의 길이는 2 이상 1,000,000 이하입니다. K는 0 이상 1,000,000,000 이하입니다. scoville의 원소는 각각 0 이상 1,000,000 이하입니다. 모든 음식의 스코빌 지수를 K 이상으로 만들 수 없는 경우에는 -1을 return 합니다. 입출력 예 설명 스코빌 지수가 1인 음식과 2인 음식을 섞으면 음식의 스코빌 지수가 아래와 같이 됩니다. 새로운 음식의 스코빌 지수 = 1 + (2 * 2) = 5 가진 음식의 스코빌 지수 = [5, 3, 9, 10, 12] 스코빌 지수가 3인 음식과 5인 음식을 섞으면 음식의 스코빌 지수가 아래와 같이 됩니다. 새로운 음식의 스코빌 지수 = 3 + (5 * 2) = 13 가진 음식의 스코빌 지수 = [13, 9, 10, 12] 모든 음식의 스코빌 지수가 7 이상이 되었고 이때 섞은 횟수는 2회입니다.