题目
输入一个整型数组,数组里有正数也有负数。数组中的一个或连续多个整数组成一个子数组。求所有子数组的和的最大值。
要求时间复杂度为O(n)。
示例:
输入: nums = [1,-2,3,10,-4,7,2,-5]
输出: 18
解释: 连续子数组 [3,10,-4,7,2] 的和最大,为18。
思路
动态规划的思想
用 f(i) 表示以第i个数字结尾的子数组的最大和,那么我们需要求的就是 max[f(i)]。
动态转移方程:
f
(
i
)
=
{
n
u
m
s
[
i
]
,
f(i-1)<=0
f
(
i
−
1
)
+
n
u
m
s
[
i
]
,
f(i-1)>0
f(i)= \begin{cases} nums[i],& \text{f(i-1)<=0}\\ f(i-1)+nums[i],& \text{f(i-1)>0} \end{cases}
f(i)={nums[i],f(i−1)+nums[i],f(i-1)<=0f(i-1)>0
当以第 i-1 个数字结尾的子数组中所有数字的和 f(i-1) 小于等于0,如果再把 nums[i] 加进来。得到的数会比 nums[i] 的值还小,所以这种情况下就把前面计算得到的 f(i-1) 舍弃,从 nums[i] 重新开始累加。
当以第 i-1 个数字结尾的子数组中所有数字的和 f(i-1) 大于0,则把第 i 个数字 nums[i] 累加进来。
实例分析
nums = [1,-2,3,10,-4,7,2,-5]
初始化时, f(0)=nums[0]=1
开始遍历:
i=1,由于f(0)>0,所以 f(1)=f(0)+nums[1]=1-2= -1;
i=2,由于f(1)<0,所以 f(2)=nums[2]=3;
i=3,由于f(2)>0,所以 f(3)=f(2)+nums[3]=3+10=13;
i=4,由于f(3)>0,所以 f(4)=f(3)+nums[4]=13-4=9;
i=5,由于f(4)>0,所以 f(5)=f(4)+nums[5]=9+7=16;
i=6,由于f(5)>0,所以 f(6)=f(5)+nums[6]=16+2=18;
i=7,由于f(6)>0,所以 f(7)=f(6)+nums[7]=18-5=13;
max{ f(i) }=f(6)=18,就是我们要求的值。
测试用例
1、功能测试(输入的数组中有正数有负数;输入的数组中全是正数;输入的数组中全是负数)。
解法
class Solution:
def maxSubArray(self, nums: List[int]) -> int:
curSum = nums[0];
GreatestSum = curSum;
for index in range(1,len(nums)):
if curSum<=0:
curSum = nums[index]
else:
curSum += nums[index]
if curSum > GreatestSum:
GreatestSum = curSum
return GreatestSum
复杂度分析
时间复杂度为O(n),空间复杂度为O(1)。