题目:POJ2479
思路:用两个数组l[i]和r[i]分别表示从左边到和从右边到i号位置的最大子段和,分别从左到右和从右到左遍历一遍即可,如果上一位置保存结果为正数,则当前最优解为上一位置最优解+当前数,否则,当前最优解为当前数(从当前位置重新开始累加,分段)(因为这时如果当前数再加上个结果相当于加了个负数,一定比当前数小),然后再遍历一遍取前缀和后缀最大为到当前位置的最大和(因为拿前缀来说,一段数可能分了多段,如果前面的结果有的比后面的大,要更新后面的结果),最后遍历每个位置两边的和,取max即为答案。
完整代码:
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn=5e4+5;
const int inf=0x3f3f3f3f;
int t,n,a[maxn],l[maxn],r[maxn];
int main()
{
// ios::sync_with_stdio(false);
// cin.tie(0);
scanf("%d",&t);
while(t--)
{
scanf("%d",&n);
//memset(a,0,sizeof(a));
for(int i=0;i<n;i++) scanf("%d",&a[i]);
l[0]=a[0];
for(int i=1;i<n;i++){
if(l[i-1]<0){
l[i]=a[i];//如果上一位置保存结果为正数,则当前最优解为上一位置最优解+当前数
}
else{
l[i]=l[i-1]+a[i];//否则,当前最优解为当前数(从当前位置重新开始累加,分段)(因为这时如果当前数再加上个结果相当于加了个负数,一定比当前数小)
}
}
for(int i=1;i<n;i++) l[i]=max(l[i],l[i-1]);//遍历一遍取前缀最大为到当前位置的最大和(因为一段数可能分了多段,如果前面的结果有的比后面的大,要更新后面的结果)
//下面,同理
r[n-1]=a[n-1];
for(int i=n-2;i>=0;i--){
if(r[i+1]<0){
r[i]=a[i];
}
else{
r[i]=r[i+1]+a[i];
}
}
for(int i=n-2;i>=0;i--) r[i]=max(r[i],r[i+1]);
int ans=-inf;
for(int i=1;i<n;i++){
ans=max(ans,l[i-1]+r[i]);
}
printf("%d\n",ans);
}
return 0;
}