Programmers Heapの方が辛いです

1534 ワード

[お尻(heap)]もうちょっと辛い
Link: https://programmers.co.kr/learn/courses/30/lessons/42626
問題の説明
辛いのが好きな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例
    scoville K return
    [1, 2, 3, 9, 10, 12] 7 2
    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回であった.
    コードアクセス
    heapを使うと、最高価格と最低価格が見つかりやすいので、popとpushの後のリストにもアクセスする必要があります.
    Code | Python
    import heapq
    def solution(scoville, K):
    	heapq.heapify(scoville)
    	answer = 0
    
    	while scoville[0] < K:
    		answer += 1
    		if len(scoville) == 1:
    			return -1
    		temp = scoville[0]
    		heapq.heappop(scoville)
    		temp += scoville[0]*2
    		heapq.heappop(scoville)
    		heapq.heappush(scoville,temp)
    		
    	return answer
    Screen Shot & Output

    Output: 2