描述
给你一个整数数组 nums ,请你找出一个具有最大和的连续子数组(子数组最少包含一个元素),返回其最大和。
子数组 是数组中的一个连续部分。
示例 1:
输入:nums = [-2,1,-3,4,-1,2,1,-5,4]
输出:6
解释:连续子数组 [4,-1,2,1] 的和最大,为 6 。示例 2:
输入:nums = [1]
输出:1示例 3:
输入:nums = [5,4,-1,7,8]
输出:23
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/maximum-subarray
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题解
这个问题可以使用动态规划来解决。我们可以定义一个变量 curSum
来表示当前的子序列和,以及一个变量 maxSum
来表示最大子序列和。我们遍历整个数组,将每个元素加到 curSum
中,并在每次加入元素后更新 maxSum
的值。如果 curSum
变成了负数,我们就将其重置为0,因为负的子序列和不可能是最大子序列和的一部分。
class Solution {
public:
int maxSubArray(vector<int>& nums) {
int maxSum = nums[0]; // 初始化最大和为数组第一个元素
int curSum = 0; // 初始化当前和为0
for (int i = 0; i < nums.size(); i++) {
curSum += nums[i]; // 累加当前元素到当前和
maxSum = max(maxSum, curSum); // 更新最大和
if (curSum < 0) { // 如果当前和小于0,重置当前和为0
curSum = 0;
}
}
return maxSum; // 返回最大和
}
};
题目还要求我们用分治法来解,这个分治法的思想就类似于二分搜索法,需要把数组一分为二,分别找出左边和右边的最大子数组之和,然后还要从中间开始向左右分别扫描,求出的最大值分别和左右两边得出的最大值相比较取最大的那一个,代码如下:
class Solution {
public:
int maxSubArray(vector<int>& nums) {
if (nums.empty()) return 0;
return helper(nums, 0, (int)nums.size() - 1);
}
int helper(vector<int>& nums, int left, int right) {
if (left >= right) return nums[left];
int mid = left + (right - left) / 2;
int lmax = helper(nums, left, mid - 1);
int rmax = helper(nums, mid + 1, right);
int mmax = nums[mid], t = mmax;
for (int i = mid - 1; i >= left; --i) {
t += nums[i];
mmax = max(mmax, t);
}
t = mmax;
for (int i = mid + 1; i <= right; ++i) {
t += nums[i];
mmax = max(mmax, t);
}
return max(mmax, max(lmax, rmax));
}
};
回顾一下,前缀和数组 preSum
就是 nums
元素的累加和,preSum[i+1] - preSum[j]
其实就是子数组 nums[j..i]
之和(根据 preSum
数组的实现,索引 0 是占位符,所以 i
有一位索引偏移)。
**那么反过来想,以 nums[i]
为结尾的最大子数组之和是多少?其实就是 preSum[i+1] - min(preSum[0..i])
**。
所以,我们可以利用前缀和数组计算以每个元素结尾的子数组之和,进而得到和最大的子数组:
class Solution {
public:
int maxSubArray(vector<int>& nums) {
int n = nums.size();
vector<int> preSum(n + 1);
preSum[0] = 0;
// 构造 nums 的前缀和数组
for (int i = 1; i <= n; i++) {
preSum[i] = preSum[i - 1] + nums[i - 1];
}
int res = INT_MIN;
int minVal = INT_MAX;
for (int i = 0; i < n; i++) {
// 维护 minVal 是 preSum[0..i] 的最小值
minVal = min(minVal, preSum[i]);
// 以 nums[i] 结尾的最大子数组和就是 preSum[i+1] - min(preSum[0..i])
res = max(res, preSum[i + 1] - minVal);
}
return res;
}
};