在Python中查找一维数组的运行总和的程序
假设我们有一个数组nums。作为rs[i]的数组的运行总和是从nums[0]到nums[i]的所有元素的总和。最后返回nums的整个运行总和。
所以,如果输入像nums=[8,3,6,2,1,4,5],那么输出将是[8,11,17,19,20,24,29],因为
rs[0] = nums[0] rs[1] = sum of nums[0..1] = 8 + 3 = 11 rs[2] = sum of nums[0..2] = 8 + 3 + 6 = 17 and so on
示例(Python)
让我们看看以下实现以获得更好的理解-
def solve(prices): n=len(nums) rs=[nums[0]] for i in range(1,n): nums[i]+=nums[i-1] rs.append(nums[i]) return rs nums = [8,3,6,2,1,4,5] print(solve(nums))
输入
[8,3,6,2,1,4,5]输出结果
[8, 11, 17, 19, 20, 24, 29]