[programmers]より辛い/python

問題の説明

辛いのが好きなLeoは全ての食べ物のScobile指数をK以上にしたいすべての食べ物のScobile指数がK以上になるように、LeoはScobile指数が最も低い2つの食べ物を以下のような特殊な方法で混合し、新しい食べ物を作った.
混合食品のスコヴェル指数=最も辛くない食品のスコヴェル指数+(第2の辛くない食品のスコヴェル指数*2)
Leoは、すべての食べ物のスコビル指数がK以上になるまでかき混ぜます.
Leoが持つ食べ物のスコヴェル指数とScoville指数がKに達したとき、すべての食べ物のスコヴェル指数をK以上に混合する最小回数を返す解関数を作成してください.

せいげんじょうけん

scovilleの長さは2以上100000以下である.

Kは0以下です.

scovilleの各要素は0または1000000未満です.

すべての食品のスコヴェル指数がKを超えない場合、−1を返す.これは優先キュー(hip)構造を利用する必要があるという問題になる.
Pythonの場合、データ構造を直接実現する必要はなく、内蔵モジュールを使用するだけでよい.

📌 説明する

import heapq as hq

def solution(scoville, K):
    hq.heapify(scoville)
    answer = 0
    
    while True:
        one = hq.heappop(scoville)
        
        if one >= K: return answer
        if not scoville: return -1
        
        two = hq.heappop(scoville)
        hq.heappush(scoville, one+two*2)
        answer += 1

学識

優先キュー挿入削除アルゴリズムを思い出した.

のあいまいな資料構造を再学習しましょう.