题目
给你一个数组 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
}
}
总结
类似的算法主要还是遵循一个原则,做过的计算要想办法利用起来,搞清楚f(n) 和 f(n-1) 的关系