求最长子段和问题,但是是求两段的子段和。
也就是对每个i来说,求出[0~i-1]的最大子段和以及[i~n-1]的最大子段和,再相加起来。
求最大的一个就行了。
[0~i-1]的最大子段和从左向右扫描,[i~n-1]从右向左扫描即可。
时间复杂度为O(n).
#include<stdio.h>
#include<iostream>
using namespace std;
int dp[50001];
int s[50001];
int main(){
int T,n,i,j;
int sum,min,ans;
scanf("%d",&T);
while(T--){
scanf("%d",&n);
sum=0;
min=INT_MIN;
for(i=0;i<n;i++){
scanf("%d",&s[i]);
sum+=s[i];
if(sum>min){
min=sum;
}
dp[i]=min;
if(sum<0){
sum=0;
}
}
sum=0;
ans=min=INT_MIN;
for(i=n-1;i>0;i--){
sum+=s[i];
if(dp[i-1]+sum>ans){
ans=dp[i-1]+sum;}
if(sum<0) sum=0;
}
printf("%d\n",ans);
}
return 0;
}
Accepted 556K 438MS C++ 581B
#include <cstdio>
#include <algorithm>
using namespace std;
int v[50001];
int left[50001];
int right[50001];
void solve()
{
int casessize;
scanf("%d",&casessize);
while(casessize--){
int size;
scanf("%d",&size);
for(int i=0;i<size;++i)
scanf("%d",&v[i]);
// 此时left[i] 为包含i最大子段和
left[0] = v[0];
for(int i=1;i<size;++i){
if(left[i-1]<0)
left[i] = v[i];
else
left[i] = left[i-1]+v[i];
}
// 此时left[i] 为i左边最大子段和
for(int i=1;i<size;++i)
left[i] = max(left[i],left[i-1]);
right[size-1] = v[size-1];
for(int j=size-2;j>=0;j--){
if(right[j+1]<0)
right[j] = v[j];
else
right[j] = right[j+1]+v[j];
}
for(int i=size-2;i>=0;i--)
right[i] = max(right[i+1],right[i]);
int res = INT_MIN;
for(int i=1;i<size;++i){
res = max(res,left[i-1]+right[i]);
}
printf("%d\n",res);
}
}
int main()
{
solve();
return 0;
}