1.题意
东东和他的女朋友(幻想的)去寿司店吃晚餐(在梦中),他发现了一个有趣的事情,这家餐厅提供的 n 个的寿司被连续的放置在桌子上 (有序),东东可以选择一段连续的寿司来吃
东东想吃鳗鱼,但是东妹想吃金枪鱼。核 平 起 见,他们想选择一段连续的寿司(这段寿司必须满足金枪鱼的数量等于鳗鱼的数量,且前一半全是一种,后一半全是另外一种)我们用1代表鳗鱼,2代表金枪鱼。
比如,[2,2,2,1,1,1]这段序列是合法的,[1,2,1,2,1,2]是非法的。因为它不满足第二个要求。
东东希望你能帮助他找到最长的一段合法寿司,以便自己能吃饱。
2.样例
Input
输入:
第一行:一个整数n(2≤n≤100000),寿司序列的长度。
第二行:n个整数(每个整数不是1就是2,意义如上所述)
Output
输出:一个整数(代表东东可以选择的最长的一段连续的且合法的寿司)
Examples
Input
7
2 2 2 1 1 2 2
Output
4
Input
6
1 2 1 2 1 2
Output
2
Input
9
2 2 1 1 1 2 2 2 2
Output
6
3.解题思路
用cnt1记录1出现的次数,cnt2记录2出现的次数,c1,c2记录1与2是否出现过,xf记录上次出现的数,当c1、c2均为1且xf与x不同时计算长度mx=max(mx,2*min(cnt1,cnt2)),还有计算长度的条件为当i=n-1时同样用此公式,计算完后要将x的相应参数初始化
4.AC代码
#include<iostream>
using namespace std;
int main()
{
int n,x;
int c1=0,c2=0;
int mx=0,cnt1=0,cnt2=0,xf=0;
cin>>n;
for(int i=0;i<n;i++)
{
cin>>x;
if(c2==0&&x==1)
{
c1=1;
cnt1++;
}
if(c1==0&&x==2)
{
c2=1;
cnt2++;
}
if(c1==1)
{
if(c2==0&&x==2)
{
c2=1;
cnt2++;
}
if(c2==1&&xf!=x)
{
mx=max(mx,2*min(cnt1,cnt2));
if(x==1)
{
cnt1=1;c1=1;
//cnt2=0;c2=0;
}
else
{
cnt2=1;c2=1;
//cnt1=0;c1=0;
}
}
if(c2==1&&xf==x)
{
if(x==1) cnt1++;
if(x==2) cnt2++;
}
}
if(c2==1)
{
if(c1==0&&x==1)
{
c1=1;
cnt1++;
}
}
if(i==n-1)
{
mx=max(mx,2*min(cnt1,cnt2));
}
xf=x;
}
cout<<mx<<endl;
}