[プログラマー]もっと辛いpython

1256 ワード
プログラマ python テキストリンク
もっと辛い
問題の説明
辛いのが好きなLeoは全ての食べ物のScobile指数をK以上にしたいすべての食べ物のScobile指数がK以上になるように、LeoはScobile指数が最も低い2つの食べ物を以下のような特殊な方法で混合し、新しい食べ物を作った.
混合食品のスコヴェル指数=最も辛くない食品のスコヴェル指数+(第2の辛くない食品のスコヴェル指数*2)
Leoは、すべての食べ物のスコビル指数がK以上になるまでかき混ぜます.
Leoが持つ食べ物のスコヴェル指数とScoville指数がKに達したとき、すべての食べ物のスコヴェル指数をK以上に混合する最小回数を返す解関数を作成してください.
せいげんじょうけん
スコビルの長さは2以上1000000以下です.
Kは0より大きく、10000000より小さい.
scovilleの各要素は0または10万以下です.
すべての食べ物のスコビル指数をKより大きくすることができない場合は、−1を返す.
毎回最小の要素を探して演算する問題です.
最小の要素を探すためにソートするよりも、お尻を利用して最小の要素をすばやく見つけたほうがいいです.
  • heapqをインポートし、少なくともhipを使用します.
    • 最小のスコビルがKを超えるまで
  • を繰り返した.
  • すべてが混合されたスコビルはKより小さいときに-1
    • に戻る
    import heapq
def solution(scoville, K):
    answer = 0
    heapq.heapify(scoville)
    # 최대 힙인 경우
    # heapq._heapify_max(scoville)
    while True:
        min1 = heapq.heappop(scoville)
        if min1 >= K:
            break
        elif len(scoville) == 0:
            answer = -1
            break
        min2 = heapq.heappop(scoville)
        sum_sc = min1 + min2 * 2
        heapq.heappush(scoville, sum_sc)
        answer += 1
    return answer
    CKD:2021のヒント、vimrc、bashrcとcheatsheet
    Google KubernetesエンジンのKubectlの最初の観察