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

1637 ワード
パイ プログラマ テキストリンク
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例


    入力:[1,2,3,9,10,12],K:7
    しゅつりょく

    に答える


    終了条件付きの繰り返し文なのでwhile文で書く方法もあります!

    コード#コード#

    import heapq

def solution(scoville, K):
    answer = 0
    # scoville을 heap의 형태로 바꿈
    heapq.heapify(scoville)
        
    # for문을 사용한 반복
    for i in range(len(scoville)-1):
    	# 반복문 종료 조건: scoville의 원소가 하나만 남거나,
        # 가장 낮은 스코빌 지수인 scoville[0]가 K보다 클 때
        if len(scoville) == 1 or scoville[0] >= K:
            break
        # 스코빌 지수를 공식에 따라 섞음
        else:
            a = heapq.heappop(scoville)
            b = heapq.heappop(scoville)
            heapq.heappush(scoville, a + (b * 2))
            answer += 1
    
    # while을 사용한 반복
    while scoville[0] < K and len(scoville) > 1:
        a = heapq.heappop(scoville)
        b = heapq.heappop(scoville)
        heapq.heappush(scoville, a + (b * 2))
        answer += 1
        
    if scoville[0] < K:
        return -1
    
    return answer
    Leetcode - 410. Split Array Largest Sum
    JAvaマルチスレッド例レコード