题目
输入一个整型数组,数组中的一个或连续多个整数组成一个子数组。求所有子数组的和的最大值。
要求时间复杂度为O(n)。
示例1:
输入: nums = [-2,1,-3,4,-1,2,1,-5,4]
输出: 6
解释: 连续子数组 [4,-1,2,1] 的和最大,为 6。
提示:
- 1 <= arr.length <= 10^5
- -100 <= arr[i] <= 100
题解
滑动窗口: 如果当前和的值小于0,那么其对于之后的子数组和是没有意义的可以丢弃,因此每次计算当前和后,都对当前和的值进行判断,若>0,就保留当前和,否则丢弃当前和。
class Solution {
public int maxSubArray(int[] nums) {
int res = -1000;
int cur = 0;
for(int i = 0; i < nums.length; i++){
int sum = cur + nums[i];
res = Math.max(res, sum);
if(sum > 0) cur = sum; // 保留当前和
else cur = 0; // 丢弃当前和
}
return res;
}
}
动态规划法
对于 nums[i] 而言,以它为结尾的子数组分两种情况:
- num[i] 自身作为独立子数组:f[i]=nums[i] ;
- num[i] 与之前的数值组成子数组,由于是子数组,其只能接在 nums[i−1],即有:
f
[
i
]
=
f
[
i
−
1
]
+
n
u
m
s
[
i
]
f[i] = f[i - 1] + nums[i]
f[i]=f[i−1]+nums[i]。
最终 f[i] 为上述两种情况取max 即可:
f [ i ] = max ( n u m s [ i ] , f [ i − 1 ] + n u m s [ i ] ) f[i] = \max(nums[i], f[i - 1] + nums[i]) f[i]=max(nums[i],f[i−1]+nums[i])
又因为当前状态f[i]只依赖于f[i-1],因此可以使用有限变量优化空间。
class Solution {
public int maxSubArray(int[] nums) {
int sum = nums[0]; // 保存前缀和
int res = sum;
for(int i = 1; i < nums.length; i++){
sum = Math.max(nums[i], sum + nums[i]); // 比较当前值和当前值加上前面子数组的最大值
res = Math.max(sum, res);
}
return res;
}
}