PythonのKプレフィックス

数値numsと整数kのリストがあるとすると、nums [0] + nums [1] + ... +nums[i]≤kである可能性のある最大のiを見つける必要があります。有効なiが存在しない場合は、-1を返します。

したがって、入力がnums =[4、-7、5、2、6]、k =5の場合、出力は3になり、4 +（-7）+ 5+2を追加した場合のインデックスは3になります。 =4、これはk未満です。最後の要素を追加すると、k以上になるため、インデックスは3になります。

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

• 範囲1からnums-1のサイズのiの場合、do
  • nums [i]：=nums [i] + nums [i-1]
• nums -1から-1の範囲サイズのiの場合、1ずつ減らします。
  • nums [i] <=kの場合、
    • 私を返す
• 戻り値-1

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

例

class Solution:
   def solve(self, nums, k):
      for i in range(1,len(nums)):
         nums[i]+=nums[i-1]
         for i in range(len(nums)-1,-1,-1):
            if nums[i]<=k:
                return i
      return -1
ob = Solution()
nums = [4, -7, 5, 2, 6]
k = 5
print(ob.solve(nums, k))

入力

[4, -7, 5, 2, 6], 5

出力

3