题目
输入一个整型数组,数组中的一个或连续多个整数组成一个子数组。求所有子数组的和的最大值。
要求时间复杂度为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
分析
采用动态规划思想做该题,首先创建一个变量max,存数组中的第一个值,变量max就是未来要求的所有子数组的和的最大值,数组从第二个元素开始遍历,再创建一个同样大小的新数组dp并将nums[0]赋值给dp[0],比较前一个值dp[i-1]加上当前值nums[i]的和与当前值nums[i]谁大,然后保存大的值赋值给dp[i],然后比较dp[i]和max的大小,更大的值赋值给max,如此反复。dp[i]表示的是以nums[i]为结尾的连续子数组的最大和。时间复杂度空间复杂度都为O(n)。
优化空间复杂度,我们发现空间上只需要dp[i-1]和dp[i]就够了,dp[i-1]顶多就加上nums[i]一起赋值给dp[i],之后遍历到下一轮就没什么作用,因此可以直接将dp变成一个变量就行了,老dp加上nums[i]赋值给新dp。
代码
class Solution {
public int maxSubArray(int[] nums) {
int[] dp = new int[nums.length];
dp[0] = nums[0];
int max = nums[0];
for(int i=1;i<=nums.length-1;i++){
dp[i] = Math.max(dp[i-1]+nums[i],nums[i]);
max = Math.max(max,dp[i]);
}
return max;
}
}
class Solution {
public int maxSubArray(int[] nums) {
int dp = nums[0];
int max = nums[0];
for(int i=1;i<=nums.length-1;i++){
dp = Math.max(dp+nums[i],nums[i]);
max = Math.max(max,dp);
}
return max;
}
}