力扣-1480.一维数组的动态和

1.题目介绍

给你一个数组 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]

2.题解

2.1 原地修改

思路

在原有数组的基础上进行修改

代码

这里一个稍微要注意的点是,由于preSum[0] = nums[0]; 所以实际开始是从i=1开始的!!!

class Solution {
public:
    vector<int> runningSum(vector<int>& nums) {
        int n = nums.size();
        for (int i = 1; i < n; i++){
            nums[i] = nums[i-1] + nums[i];
        }
        return nums;
    }
};

2.2 一维前缀和

思路

使用一个前缀和数组来维护preSum[i-1],而preSum[i] = preSum[i-1] + nums[i]

代码

class Solution {
public:
    vector<int> runningSum(vector<int>& nums) {
        vector<int> preSum;
        int sum = 0;
        for (auto num:nums){
            sum += num;
            preSum.push_back(sum);
        }
        return preSum;
    }
};
posted @ 2024-01-18 16:41  DawnTraveler  阅读(13)  评论(0编辑  收藏  举报