1480. 一维数组的动态和
给你一个数组 nums 。数组「动态和」的计算公式为:runningSum[i] = sum(nums[0]…nums[i]) 。
请返回 nums 的动态和。
示例 1:
输入:nums = [1,2,3,4]
输出:[1,3,6,10]
解释:动态和计算过程为 [1, 1+2, 1+2+3, 1+2+3+4] 。
示例 2:
输入:nums = [1,1,1,1,1]
输出:[1,2,3,4,5]
解释:动态和计算过程为 [1, 1+1, 1+1+1, 1+1+1+1, 1+1+1+1+1] 。
示例 3:
输入:nums = [3,1,2,10,1]
输出:[3,4,6,16,17]
思路
sum(nums[0]…nums[i])
即,求前 i 项和。 => 前缀和
本题前缀和首项不为 0, prefixSum[0] = nums[0]
代码实现可参考构造前缀和数组代码。
最终代码
class Solution {
public int[] runningSum(int[] nums) {
if(nums == null || nums.length == 0) {
return null;
}
int[] prefixSum = new int[nums.length];
for(int i = 0; i < nums.length; i++) {
prefixSum[i] = (i == 0) ? nums[0] : prefixSum[i - 1] + nums[i];
}
return prefixSum;
}
}