在 Python 中为所有正前缀和找到要在开头插入的最小值的程序
假设我们有一个名为nums的数字列表。我们必须找到可以在nums开头插入的最小正值,以便结果列表的前缀和包含全部大于0的数字。
所以,如果输入像nums=[3,-6,4,3],那么输出将是4,因为如果我们插入4到列表中,那么我们有[4,3,-6,4,3].现在前缀和是[4,7,1,5,8],都大于0。
示例
让我们看下面的实现来更好地理解
def solve(nums): nums.insert(0, 0) for i in range(1, len(nums)): nums[i] += nums[i - 1] return 1 - min(nums) nums = [3, -6, 4, 3] print(solve(nums))
输入
[3, -6, 4, 3]输出结果
4