剑指 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
解题思路:
首先判断给定的nums数组的长度,如果为0,则没必要继续操作,直接返回0
如果不为0,则继续:
**状态定义:**设动态规划列表dp,dp[i]代表以元素nums[i]为结尾的连续子数组最大和
为何定义最大和 dp[i]dp[i] 中必须包含元素 nums[i]nums[i] :保证 dp[i]dp[i] 递推到 dp[i+1]dp[i+1] 的正确性;如果不包含 nums[i]nums[i] ,递推时则不满足题目的 连续子数组 要求。
转移方程: 若 dp[i-1] \leq 0dp[i−1]≤0 ,说明 dp[i - 1]dp[i−1] 对 dp[i]dp[i] 产生负贡献,即 dp[i-1] + nums[i]dp[i−1]+nums[i] 还不如 nums[i]nums[i] 本身大。
-
当 dp[i - 1] > 0 时:执行 dp[i] = dp[i-1] + nums[i];
-
当 dp[i - 1] ≤0 时:执行 dp[i] = nums[i];
class Solution {
public:
int maxSubArray(vector<int>& nums) {
int len = nums.size();
if(len == 0) return 0;
int dp[len];
dp[0] = nums[0];
int res =dp[0] ;
for(int i = 1; i < len; i++){
dp[i] = max(dp[i - 1] + nums[i],nums[i]);
res = max(res,dp[i]);
}
return res;
}
};
利用滚动数组思想进一步优化
class Solution {
public:
int maxSubArray(vector<int>& nums) {
if(nums.size() == 0) return 0;
int pre = nums[0];
int res = nums[0];
for(int i = 1; i < nums.size(); i++){
pre = max(pre+nums[i], nums[i]);
res = max(pre, res);
}
return res;
}
};