LeetCode题解之一维数组的动态和

一维数组的动态和

题目描述

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

解题思路

使用动态规划，从题目中很容易就可以得知状态转移方程为：dp[0] = nums[0]，dp[i] = dp[i-1]+nums[i](i >= 1)。

复杂度分析

• 时间复杂度：O(n)。
• 空间复杂度：O(1)。

代码实现

class Solution {
    public int[] runningSum(int[] nums) {
        int[] res = new int[nums.length];
        res[0] = nums[0];
        for (int i = 1; i < nums.length; i ++) {
            res[i] = res[i-1] + nums[i];
        }
        return res;
    }
}