LeetCode刷题 - #1480. 一维数组的动态和

67 阅读1分钟

1480. 一维数组的动态和

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

思路

sum(nums[0]…nums[i])即,求前 i 项和。 => 前缀和

本题前缀和首项不为 0, prefixSum[0] = nums[0]

代码实现可参考构造前缀和数组代码。

最终代码

class Solution {
    public int[] runningSum(int[] nums) {
        if(nums == null || nums.length == 0) {
            return null;
        }
        int[] prefixSum = new int[nums.length];
        for(int i = 0; i < nums.length; i++) {
            prefixSum[i] = (i == 0) ? nums[0] : prefixSum[i - 1] + nums[i];
        }
        return prefixSum;
    }
}