题目描述
输入一个整型数组,数组中的一个或连续多个整数组成一个子数组。求所有子数组的和的最大值。
要求时间复杂度为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
解题思路 - 动态规划
首先,不知道为啥发现04-30没有打卡,我都蒙了。然后又找不到草稿,只好重写一遍。也因为是做过的,所以很快就写出来了。权当复习吧。
// 由于题目限制为连续的子数组
// f(i)为以nums[i]元素结尾的最大连续和。
// 那么则有 f(i) = max(f(i-1)+nums[i], 0);
// 因为如果前面的数比0还小,那么最大的子数组和肯定是当前元素
第一次做的时候怎么想的居然忘记了。但目测没做出来,因为没有看到LeetCode的提交记录。当时应该是看了官方的题解吧。
突破口:
1. 连续子数组
2. 数学规律:求和最大时,前面若是负数和,则为负增益,还不如从当前元素重新找最大值。就好比,如果全部都是负数,那么有最大和的子数组一定只有一个元素。
解题代码
public int maxSubArray(int[] nums) {
// 补2021-04-30 动态规划
// 由于题目限制为连续的子数组
// f(i)为以nums[i]元素结尾的最大连续和。
// 那么则有 f(i) = max(f(i-1)+nums[i], 0);
// 因为如果前面的数比0还小,那么最大的子数组和肯定是当前元素
int[] dp = new int[nums.length];
dp[0] = nums[0];
// 初始条件
int max = nums[0];
for (int i=1; i<nums.length; i++) {
dp[i] = nums[i];
if (dp[i-1] > 0) {
dp[i] += dp[i-1];
}
max = max < dp[i] ? dp[i] : max;
}
return max;
}