Programmers Coding Quiz#37もっと辛い!

問題の説明

辛いのが好きな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を返す.

I/O例

scovilleKreturn[1, 2, 3, 9, 10, 12]72

I/O例説明

Scobile指数が1の食品と2の食品を混合し、食品のScobile指数は以下の通りである.
新しい食べ物のスコヴェル指数=1+(2*2)=5
食物を持つスコヴェル指数=[5,3,9,10,12]

スコヴェル指数が3の食べ物と5の食べ物を混合し、食べ物のスコヴェル指数は以下の通りである.
新しい食べ物のスコヴェル指数=3+(5*2)=13
食物を持つスコヴェル指数=[13,9,10,12]

すべての食品のスコビル指数は7以上で,このときの攪拌回数は2回であった.

に答える

import heapq

def solution(scoville, K):
    result = []
    
    for v in scoville:
        heapq.heappush(result, v)
    
    count = 0

    while result[0] < K:
        heapq.heappush(result, heapq.heappop(result) + (heapq.heappop(result) * 2))

        count += 1

        if len(result) == 1 and result[0] < K:
            count = -1
            break
            
    return count

heapqモジュールを使用してscovilleをリストhipに変換します.
heapqモジュールを使用して、生成された空のリストに要素heapqを配置します.プッシュで送ると、一番小さいお尻が使えます.
配列をpopまたはpushで置き換える場合は、昇順で並べ替えることもできます.whileは結果[0]=最小値がKより小さい場合、heapushとheappopを使用して新しい要素の除去と追加を同時に行います.
removeとは異なり、popは「カット」に近いため、計算値の要素として使用できます.
最小値の削除->2番目の最小値の削除->合計値の追加
この3つのステップの後、countは1増加します.whileサイクルが終了すると、累積したcountが答えになります.
もちろん、ここにも問題条件の例外があります.
リスト内のすべての要素(問題条件など)を計算中に値を上げ続けると、Kの値に達することはできません.[1,2,3]K=15に示すように、1番目の1 + 2 * 2 = 5が2番目の3 + 5 * 2であれば、すべての元素が使い切れてもKには達しない.この場合、演算終了後に1つの要素しか残っていないがK未満の場合、countを-1として指定し、while文を強制的に閉じることができます.