Python中的K前缀
假设我们有一个数字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
输出结果
3