剑指 Offer 42. 连续子数组的最大和
输入一个整型数组,数组里有正数也有负数。数组中的一个或连续多个整数组成一个子数组。求所有子数组的和的最大值。
要求时间复杂度为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
(来源:力扣(LeetCode))
思路:动态规划
首先分解子问题:
因为子数组是一段连续的整数,所以可以由一个起始点和一个终止点确定一个子数组。我们令一个子数组的截止元素在i这个位置,此时最大和有两种情况:
1、i这个位置的元素就是一个子数组;
2、i这个位置的元素的值 + 以 i- 1 结尾的所有子数组中的子数组和最大的值。
状态方程:
我们设置一个dp数组,且dp[i]表示的是“以i结尾的所有子数组的最大值”。
通过上述的情况分析,我们可以得到状态转移方程:
dp[i] = Math.max(dp[i - 1] + nums[i], nums[i])
简化为:
dp[i] = Math.max(dp[i - 1], 0) + nums[i]
代码实现:
1、若数组长度为0或数组为空,则返回0;
2、创建dp数组,且初始化为dp[0] = nums[0];
3、再设置一个变量res,负责记录子数组的最大值,因为子数组有可能以数组中的任意一个元素结尾,初始化为res = dp[0];
4、遍历数组nums,利用状态方程得到每段子数组的最大值,并将最后结果记录进res;
5、返回res即可。
class Solution {
public int maxSubArray(int[] nums) {
if (nums.length == 0 || nums == null) {
return 0;
}
int n = nums.length;
int[] dp = new int[n + 1];
dp[0] = nums[0];
int res = dp[0];
for (int i = 1; i < n; ++i) {
dp[i] = Math.max(dp[i - 1], 0) + nums[i];
res = Math.max(res, dp[i]);
}
return res;
}
}