最大子段和
题目描述
给出一个长度为 n n n 的序列 a a a,选出其中连续且非空的一段使得这段和最大。
输入格式
第一行是一个整数,表示序列的长度 n n n。
第二行有 n n n 个整数,第 i i i 个整数表示序列的第 i i i 个数字 a i a_i ai。
输出格式
输出一行一个整数表示答案。
样例 #1
样例输入
7
2 -4 3 -1 2 -4 3
样例输出
4
提示
样例 1 解释
选取 [ 3 , 5 ] [3, 5] [3,5] 子段 { 3 , − 1 , 2 } \{3, -1, 2\} {3,−1,2},其和为 4 4 4。
数据规模与约定
- 对于 40 % 40\% 40% 的数据,保证 n ≤ 2 × 1 0 3 n \leq 2 \times 10^3 n≤2×103。
- 对于 100 % 100\% 100% 的数据,保证 1 ≤ n ≤ 2 × 1 0 5 1 \leq n \leq 2 \times 10^5 1≤n≤2×105, − 1 0 4 ≤ a i ≤ 1 0 4 -10^4 \leq a_i \leq 10^4 −104≤ai≤104。
题解1:暴力枚举
思路
我们用两个循环,分别枚举不同子段的头和尾。再用第三个循环将头尾之间的数相加,得到子段和。然后将每次得到的和与上一次的相比,留下最大的作为答案。
代码
#include<bits/stdc++.h>
using namespace std;
int n,a[200005],sum,ans=-10005;//ans的初值一定要定的比所有数都小,这样才能避免比较时没有记录答案的情况(这道题里的最小值是−10的4次方)
int main()
{
cin>>n;//输入一共有几个数
for(int i=1;i<=n;i++)cin>>a[i];///分别输入这些数
for(int i=1;i<=n;i++)//第一层循环,枚举子段头
{
for(int j=i;j<=n;j++)//第二层循环,枚举子段尾。因为尾不可以在头的前面,所以j=i
{
sum=0;//每次计算和之前清零sum
for(int k=i;k<=j;k++)//将从头到尾的数相加
{
sum+=a[k];
}
ans=max(ans,sum);//与上次进行比较,留下最大的当答案
}
}
cout<<ans;//输出
return 0;
}
优化
由于每次都要单独计算子段和,非常浪费时间,我们就可以用前缀和算法优化掉。
前缀和算法就是先算出从第一个数到每一个数的和,再用特定的和相减算出不同子段和的算法。
拿样例举例,从第一到第三个数的和为1,从第一到第六个数的和为-2。而用-2减去1,得到-3,正好是子段4-6的和。
由此可知,子段和等于它的尾的前缀和减去头前面一个位置的前缀和。
代码
#include<bits/stdc++.h>
using namespace std;
const int x=200005;
int n,a[x],f[x],sum,ans=-10005;
int main()
{
cin>>n;
for(int i=1;i<=n;i++)cin>>a[i];
for(int i=1;i<=n;i++)//算出所有前缀和
{
f[i]=f[i-1]+a[i];//每个前缀和都等于它的上一位加上当前的数字
}
for(int i=1;i<=n;i++)//枚举头
{
for(int j=i;j<=n;j++)//枚举尾
{
sum=f[j]-f[i-1];//用尾前缀和减头前一位前缀和,得到当前子段和
if(sum>ans)ans=sum;
}
}
cout<<ans;
return 0;
}
题解2:动态规划
思路
先找出小子段的最大子段和,再一步步将小子段增长,每次记录最大的和,最终得到全段的最大子段和。
详解
众所周知,最大的加最大的得到的和肯定最大,在求最大子段和的时候也是一样。长度最小子段的最大子段和只能是自己,于是记录下来。而将子段长+1后,也就是加入一个新数后,这时有三种选择:新数加入之前的最大子段,新数单独作一个新子段,保留之前最大子段为答案。然后只需选选项中最大的当答案即可。
代码
#include<bits/stdc++.h>
using namespace std;
int n,a,f=0,ans=-100000000;
int main()
{
cin>>n;//输入总数
for(int i=1;i<=n;i++)//分别输入并求最大子段
{
cin>>a;//具体数字
if(f+a>a)f=f+a;//如果将新数加入之前最大子段,总和比新数自身大,那么加入新数
else f=a;//反之,如果新数更大,将新数单独作为新子段
ans=max(ans,f);//比较当前最大子段和上一次的最大子段,留下最大的
}
cout<<ans;//输出
return 0;
}