PythonでK個のサブリストを増やした後に最小値を最大化するようにプログラムする

numsと呼ばれる数値のリストと2つの値、sizeとkがあるとします。ここで、長さサイズの連続したサブリストを取得し、すべての要素を1つずつインクリメントする操作があるとします。この操作はk回実行できます。可能な限り最大の最小値を、numsで見つける必要があります。

したがって、入力がnums =[2、5、2、2、7]、size =3、k =2の場合、[2、5、2]を増やして[を取得できるため、出力は3になります。 3、6、3、2、7]次に[6、3、2]をインクリメントして、[3、7、4、3、7]を取得します。最小値は3

これを解決するには、次の手順に従います-

• 関数possible（）を定義します。これはターゲットになります
• events：=サイズNのリストで、0で埋めます
• 移動：=0、s：=0
• 0からNの範囲のiについては、
  • s：=s+イベント[i]
  • デルタ：=ターゲット-（A [i] + s）
  • デルタ>0の場合、
    • 移動：=移動+デルタ
    • s：=s+デルタ
    • i +サイズ
    • events [i + size]：=events [i + size] --delta

理解を深めるために、次の実装を見てみましょう-

例

class Solution:
   def solve(self, A, size, K):
      N = len(A)
   def possible(target):
      events = [0] * N
      moves = s = 0
      for i in range(N):
         s += events[i]
         delta = target - (A[i] + s)
         if delta > 0:
            moves += delta
            s += delta
            if i + size < N:
               events[i + size] -= delta
               return moves <= K
               left, right = 0, 10 ** 10
               while left < right:
                  mid = (left + right + 1)//2
               if possible(mid):
                  left = mid
               else:
                  right = mid - 1
      return left
ob = Solution()
nums = [2, 5, 2, 2, 7]
size = 3
k = 2
print(ob.solve(nums, size, k))

入力

[2, 5, 2, 2, 7], 3, 2

出力

3