[算法学习] LeetCode 1480. 一维数组的动态和

最新推荐文章于 2023-07-19 21:28:28 发布

阿飞正传的阿飞

最新推荐文章于 2023-07-19 21:28:28 发布

阅读量171

点赞数

分类专栏：算法学习文章标签：算法 leetcode 学习

本文链接：https://blog.csdn.net/njafei/article/details/126233359

版权

算法学习专栏收录该内容

4 篇文章 1 订阅

订阅专栏

本文讨论了一种动态求和的算法问题，通过示例展示了如何计算数组的动态和，并对比了传统解法与优化后的代码。优化后的代码利用了前缀和的思想，提高了效率，降低了空间复杂度。核心思想是在已有的计算基础上复用结果，避免重复计算。

摘要由CSDN通过智能技术生成

题目

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

思路

这种类似的问题，都可以归纳斐波那契数列问题，要在传统的暴力解法的基础上，做一些数据的存储和复用，这题的思路也比较简单，关系如下：

f(n) = f(n-1) + n[n]

解题

我开始写了最传统的斐波那契数组的代码，如下：

class Solution {
    func runningSum(_ nums: [Int]) -> [Int] {
        var sum = 0
        var result : [Int] = []
        
        for i in 0..<nums.count {
            sum += nums[i]
            result.append(sum)
        }
        return result
    }
}

优化

上述的代码在时间复杂度没有问题，是O(n)，空间复杂度是O(n-1) , 但是我看了下其他人的解法，比我写的更优雅基于原数组来做改，如下：

class Solution {
    func runningSum(_ nums: [Int]) -> [Int] {
        var sum = nums

        for i in 1..<nums.count {
            sum[i] += sum[i-1]
        }
        return sum
    }
}