题目
给你一个数组 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]
思路和算法
声明一个数组sums[]表示数组nums的动态和,根据动态和的定义,sums[i]=nums[0,i]中所有元素之和,因此可以直接计算动态和
var runningSum = function(nums) {
let length = nums.length;
let sums = new Array(length).fill(0);
for (let i = 0; i < length; i++) {
for (let j = 0; j <= i; j++) {
sums[i] += nums[j]
}
}
return sums;
};
输入[1,2,3,4,5]
预期输出[1,3,6,10,15]
实际输出[1,3,6,10,15]
加强1
通过重新检查我们发现
数组的第一项永远是相等的,并且sums[i]的值等于sums[i-1]+nums[i](因为sums[i-1]就是nums[i]前几项的和
因此我们可以修改算法为
var runningSum = function(nums) {
let length = nums.length;
let sums = new Array(length).fill(0);
sums[0] = nums[0];
for (let i = 1; i < length; i++) {
sums[i] = nums[i] + sums[i - 1]
}
return sums;
};
输入[1,2,3,4,5]
预期输出[1,3,6,10,15]
实际输出[1,3,6,10,15]
加强2
既然前n项的和是n-1,那为什么不直接修改元素组,毕竟这样看来原数组的前n项用不上了,毕竟修改后的n-1项就是原前n项的和,所以不存在数据丢失
for (let i = 1; i < nums.length; i++) {
nums[i] += nums[i-1];
}
return nums
最终代码如上