题目描述
给定一个整数数组 nums ,找到一个具有最大和的连续子数组(子数组最少包含一个元素),返回其最大和。
输入: [-2,1,-3,4,-1,2,1,-5,4],
输出: 6
解释: 连续子数组 [4,-1,2,1] 的和最大,为 6。
设 nums 数组的长度是 n,下标从 0 到 n-1。用 f ( i ) f(i) f(i)表示以第 i i i个数「结尾的连续子数组的最大和」,那么我们要求的答案是: max 0 ≤ i ≤ n − 1 { f ( i ) } \max \limits_{0\leq i\leq n-1}\left\{f(i)\right\} 0≤i≤n−1max{f(i)}
因此我们只需要求出每个位置的 f ( i ) f(i) f(i),然后找到 f ( i ) f(i) f(i)数组中的最大值即可。此时问题转换为如何求 f ( i ) f(i) f(i)。nums[i]单独成为一段还是加入 f ( i − 1 ) f(i-1) f(i−1)对应的那一段,这取决于nums[i]和 f ( i − 1 ) + n u m s [ i ] f(i-1)+nums[i] f(i−1)+nums[i]的大小,我们希望获得一个比较大的,于是DP转移方程: f ( i ) = max { f ( i − 1 ) + n u m s [ i ] , n u m s [ i ] } f(i)=\max\left\{f(i-1)+nums[i],nums[i]\right\} f(i)=max{f(i−1)+nums[i],nums[i]}
用一个 f f f数组来保存 f ( i ) f(i) f(i)的数值,用一个循环求出所有的 f ( i ) f(i) f(i)。因为 f ( i ) f(i) f(i)至于 f ( i − 1 ) f(i-1) f(i−1)有关,于是可以只用一个变量来维护当前 f ( i ) f(i) f(i)的 f ( i − 1 ) f(i-1) f(i−1)的数值是多少。或者直接修改原数组。
#直接修改原数组空间复杂度O(1) 原地算法
class Solution:
def maxSubArray(self, nums: List[int]) -> int:
# 空间复杂度O(1)
for i in range(1,len(nums)):
nums[i] = max(nums[i],nums[i-1]+nums[i])
return max(nums)
上面的代码的
f
(
i
)
f(i)
f(i)的定义表示以第
i
i
i个数「结尾的连续子数组的最大和,这里我们定义
f
(
i
)
f(i)
f(i)是到第
i
i
i 项为止的最大子序列和。如果 nums[i-1]大于0的话,新的nums[i]就是和前一项的和,否则就是自身。
状态转移的过程如下
1.初始情况:dp[0] = nums[0]
2.若 nums[i] > 0,那么 dp[i] = nums[i] + dp[i - 1]
3.若 nums[i] <= 0,那么 dp[i] = nums[i]
其实也是贪心算法:若当前所有元素之和小于0,则丢弃当前元素之前的数列。
class Solution:
def maxSubArray(self, nums: List[int]) -> int:
# nums[i-1]:到前一项为止的最大子序列和
for i in range (1,len(nums)):
nums[i] = nums[i] + max(0,nums[i-1])
return max(nums)