目录
题目描述:
给定一个整数数组 nums
,找到一个具有最大和的连续子数组(子数组最少包含一个元素),返回其最大和。
方法一:
class Solution {
public:
int MAX = -999999;
int sum = 0;
int maxSubArray(vector<int>& nums) {
for (int i = 0; i < nums.size(); i++)
{
sum = nums[i];
if(sum > MAX)
{
MAX = sum;
}
for (int j = i + 1; j < nums.size(); j++)
{
if (sum + nums[j] > MAX)
{
MAX = sum + nums[j];
}
sum += nums[j];
}
}
return MAX;
}
};
第一时间想到的最简单无脑的方法,时间复杂度为O(n²)。
方法二(动态规划法):
//f(i) = max(f(i-1)+nums[i] , nums[i])
class Solution {
public:
int maxSubArray(vector<int>& nums) {
//这里不能只用一个pre变量,然后最后返回pre,因为pre记录的是以nums[i]为结尾的子序列的最大和,
//所以最终得到的pre是以最后一个元素为结尾的子序列的最大和,这不一定是整个nums中的最大子序列和
int maxnums = nums[0]; //存储最大的f(i)
int pre = nums[0]; //实时记录f(i-1)
for (int i = 1; i < nums.size(); i++)
{
pre = max(pre + nums[i], nums[i]);
maxnums = max(maxnums , pre);
}
return maxnums;
}
};
设f(i)为以nums[i]为结尾(重点,即该连续子数组必须类似为i-3,i-2,i-1,i或者i)的连续子数组的最大和, 则可推出递推关系式:f(i) = max(f(i-1)+nums[i] , nums[i]),利用该关系式求出所有的f(i),题目所要求的即是最大的那个f(i)。
时间复杂度为O(n),利用一个pre变量来保存f(i-1)可将空间复杂度减少为O(1)。
方法三(前缀和法):
来自2022.8.12
class Solution {
public:
int maxSubArray(vector<int>& nums) {
int n = nums.size();
vector<int> presum(n, 0); //存储前缀和
presum[0] = nums[0];
for (int i = 1; i < n; i++)
{
presum[i] = presum[i - 1] + nums[i];
}
int ans = presum[0]; //实时记录最大子数组和
int minpresum = 0; //从前往后,最小的前缀和
for (int i = 0; i < n; i++)
{
//presum[i] - minpresum就是以nums[i]为结尾的子序列的最大和
if (presum[i] - minpresum > ans)
{
ans = presum[i] - minpresum;
}
//为什么最小的前缀和要放在上面的计算之后才更新呢?因为如果最小前缀和是自身的话,一减不就变成0了?
//实际上presum[i] - presum[i-1]得到的才是nums[i]的值,所以minpresum不能提前更新
minpresum = min(minpresum, presum[i]); //更新最小的前缀和
}
return ans;
}
};