Python中的K前缀

2026-06-04 6 花语

假设我们有一个数字nums和一个整数k的列表，我们必须找到最大可能的i，其中nums[0]+nums[1]+...+nums[i]≤k。如果不存在有效的i，我们将返回-1。

因此，如果输入像nums=[4，-7，5，2，6]，k=5，则输出将为3，索引为3，就像我们加4+(-7)+5+2=4，它小于k，如果我们添加最后一个元素，它将不再小于k，因此索引为3。

为了解决这个问题，我们将遵循以下步骤-

对于范围从1到nums的i-1，执行

nums[i]：=nums[i]+nums[i-1]

对于范围为-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

输出结果