環狀最大兩段子段和

https://www.luogu.org/problem/P1121

這個題非常有意思。有2種情況,一種是不用首尾相連的方法,另一種是要首尾相連的。首先考慮第一種,應該還是挺好寫的吧。。。就是直接算一下。用f[i]表示以第一個數開頭到i的最大子段和,g[i]就是表示以最後一個數結尾的到i的最大子段和。

然後另外一種情況的話,我們就可以在整個子段中去掉最小的兩個子段,然剩下的三段就是最大的了。

2個取一個max

但是要注意的是:如果只有一個正數的話,第二種情況就不成立(自己想),我們要直接輸出他自己。

#include <bits/stdc++.h>

using namespace std;
const int maxn=10005;
const int Inf=0x3f3f3f3f;
int n,a[maxn],f[maxn],g[maxn]; 
int query()
{
	int res=-Inf;
	for(int i=1;i<=n;i++) f[i]=max(f[i-1]+a[i],a[i]);
	for(int i=1;i<=n;i++) f[i]=max(f[i],f[i-1]);
	for(int i=n;i>=1;i--) g[i]=max(g[i+1]+a[i],a[i]);
	for(int i=n;i>=1;i--) g[i]=max(g[i],g[i+1]);
	for(int i=1;i<n;i++) res=max(res,f[i]+g[i+1]);
	return res;
}
int main()
{
	int sum=0;
	scanf("%d",&n);int tot=0;
	for(int i=1;i<=n;i++) 
	{
		scanf("%d",&a[i]);
		sum+=a[i];	
		tot+=(a[i]>0);
	}
	memset(f,-Inf,sizeof(f));
	memset(g,-Inf,sizeof(f));
	int t1=query();
	if(tot==1) printf("%d",t1);
	else 
	{
		for(int i=1;i<=n;i++) a[i]*=-1;
		int t2=sum+query();
		if(!t2) t2=-Inf;
		printf("%d",max(t1,t2));
	}
	return 0;
} 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章