吃寿司

问题描述

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

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

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

东东希望你能帮助他找到最长的一段合法寿司，以便自己能吃饱。
【输入】
第一行：一个整数n（2≤n≤100000），寿司序列的长度。
第二行：n个整数（每个整数不是1就是2，意义如上所述）
【输出】
一个整数（代表东东可以选择的最长的一段连续的且合法的寿司）
【样例输入】
9
2 2 1 1 1 2 2 2 2
【样例输出】
6

问题分析

设定两个栈，在这里我们使用map来表示栈，一个表示1，一个表示2，下面解释压栈规则，使用样例进行解释：221112222
mp【1】是压1的栈，mp【2】是压2的栈
对于第一个2，前面没有其他数，因此直接入栈mp【2】，下一个2，与前一个数相同，直接入栈
mp【1】
mp【2】2 2
当第三个数出现1的时候，与前面的数不同，统计两个栈里的答案
*ans=min(stack1.size(),stack2.size())2=0;
加入1，连续三个，不发生改变，直接入栈
mp【1】1 1 1
mp【2】2 2
当第6个数入栈的时候与第五位的1不同则计算答案并对栈清空
*ans=min(stack1.size(),stack2.size())2=4;
清空2的栈后进行后续放置最后
mp【1】1 1 1
mp【2】2 2 2 2
最后答案为
*ans=min(stack1.size(),stack2.size())2=6;

#include<iostream>
#include<map>
using namespace std;
int a[100010];
map<int,int> mp;//栈元素 
int main()
{
	int n;
	cin>>n;
	int pre=0,ans=0; 
	for(int i=1; i<=n; i++)
		cin>>a[i];
	for(int i=1;i<=n;i++)
	{
		if(pre==0)//第一个元素
		{
			mp[a[i]]++;
			pre=a[i]; 
		} 
		else
		{
			if(a[i]==pre)
				mp[a[i]]++;
			else//改变 
			{
			
				ans=max(ans,2*min(mp[1],mp[2])); 
				mp[a[i]]=1;//重新加入 
				pre=a[i]; 
			}	
		}
	}
	ans=max(ans,2*min(mp[1],mp[2])); 
	cout<<ans<<endl;
}