1、题目1480
给你一个数组 nums 。数组「动态和」的计算公式为:runningSum[i] = sum(nums[0]…nums[i]) 。
请返回 nums 的动态和。
示例:
输入:nums = [1,2,3,4]
输出:[1,3,6,10]
解释:动态和计算过程为 [1, 1+2, 1+2+3, 1+2+3+4] 。
2、求解
语言:C++
根据题目,可以知道该题是对数组中的元素累加求和。
2.1、解法一
通过一个中间变量 sum,记录第 i 个元素的求和结果,当 i+1 时,只需再加上 nums[i+1] 即可。
class Solution {public: vector runningSum(vector& nums) { int sum = 0; vector result; for(const auto& i : nums) { sum += i; result.push_back(sum); } return result; }};
这里使用了基于范围的 for 循环。
2.2、解法二
上述代码中,push_back的作用是把sum添加到在result中。但当result大小超过当前vector容量时,会重新自动分配新的存储空间。
而这里,我们知道 result 和 输入 nums 的大小是一致的,因此,为了避免重新分配内存带来的问题,可以通过 resize() 提前设定好 result 的大小。代码如下:
class Solution {public: vector runningSum(vector& nums) { int sum = 0; vector result; result.resize(nums.size()); for(int i = 0; i < nums.size(); ++i) { sum += nums[i]; result[i] = sum; } return result; }};
2.3、解法三
这里还有一种写法,由于函数入参 nums 没有 const 修饰,若可以修改入参 nums,则可以直接修改 nums,并返回 nums。
class Solution {public: vector runningSum(vector& nums) { for(int i = 1; i < nums.size(); ++i) { nums[i] = nums[i-1] + nums[i]; } return nums; }};
这里若 nums 的 size 为 0 或 1,则不会进入循环,直接返回 nums,因此不需要单独处理 nums 的 size 为 0 或 1 的情况。
在实际工作中,若可以修改入参 nums,则函数就不需要有返回值;若不可以修改入参 nums,则需要加上 const 修饰,防止误修改。