程序设计思维与实践 Week11 作业 D 必做题11-4

题目描述：

东东和他的女朋友(幻想的)去寿司店吃晚餐(在梦中)，他发现了一个有趣的事情，这家餐厅提供的 n 个的寿司被连续的放置在桌子上 (有序)，东东可以选择一段连续的寿司来吃

东东想吃鳗鱼，但是东妹想吃金枪鱼。核 平 起 见，他们想选择一段连续的寿司（这段寿司必须满足金枪鱼的数量等于鳗鱼的数量，且前一半全是一种，后一半全是另外一种）我们用1代表鳗鱼，2代表金枪鱼。

比如，[2,2,2,1,1,1]这段序列是合法的，[1,2,1,2,1,2]是非法的。因为它不满足第二个要求。

东东希望你能帮助他找到最长的一段合法寿司，以便自己能吃饱。

input：

输入：
第一行：一个整数n（2≤n≤100000），寿司序列的长度。
第二行：n个整数（每个整数不是1就是2，意义如上所述）

output：

输出：一个整数（代表东东可以选择的最长的一段连续的且合法的寿司）

思路：

前缀和处理一下，原序列为a，前缀和序列为b，bi表示以数字ai为尾部，已经持续了多少个了，例如：a1~a5为：1 1 1 2 1，则b1==1，即：以1为尾部，持续了1个。b3==3，即：以1为尾部，持续了3个。b4==1，即：以2为尾部，持续了一个。

之后开始遍历整个序列，若ai为1，则判断a[i-b[i]]是否为2，且b[i-b[i]]>=b[i]（看一下2的个数是不是比1的多）是否成立，若成立，则ans=max(ans,b[i]);

若ai为2，同理。

#include<iostream>
using namespace std;
int n,ans,a[100010],b[100010];
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
		if(a[i]==a[i-1])
		b[i]=b[i-1]+1;
		else b[i]=1;
	}
	for(int i=1;i<=n;i++)
	if(a[i]==1)
	{
		if(a[i-b[i]]==2&&b[i-b[i]]>=b[i])
		ans=max(ans,b[i]);
	}
	else
	{
		if(a[i-b[i]]==1&&b[i-b[i]]>=b[i])
		ans=max(ans,b[i]);
	}
	cout<<ans*2;
	return 0;
}