标签:数组、前缀和
题目链接:
题目描述:
给你一个数组 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]
提示:
1 <= nums.length <= 1000
-10^6 <= nums[i] <= 10^6
解题思路:
新的数组长度和原数组长度一样。第一个元素一定是原数组的第一个值。
class Solution {
public int[] runningSum(int[] nums) {
int[] result = new int[nums.length];
result[0] = nums[0];
int sum = result[0];
for(int i=1;i<nums.length;i++){
result[i] = nums[i] + sum;
sum = result[i];
}
return result;
}
}
方法2:原地修改
不需要重新new一个数组。每次就将当前索引前的数值都做和操作,只需要将当前值加进去就可以了。
class Solution {
public int[] runningSum(int[] nums) {
int n = nums.length;
for (int i = 1; i < n; i++) {
nums[i] += nums[i - 1];
}
return nums;
}
}