题意:
东东和他的女朋友(幻想的)去寿司店吃晚餐(在梦中),他发现了一个有趣的事情,这家餐厅提供的 n 个的寿司被连续的放置在桌子上 (有序),东东可以选择一段连续的寿司来吃
东东想吃鳗鱼,但是东妹想吃金枪鱼。核 平 起 见,他们想选择一段连续的寿司(这段寿司必须满足金枪鱼的数量等于鳗鱼的数量,且前一半全是一种,后一半全是另外一种)我们用1代表鳗鱼,2代表金枪鱼。
比如,[2,2,2,1,1,1]这段序列是合法的,[1,2,1,2,1,2]是非法的。因为它不满足第二个要求。
东东希望你能帮助他找到最长的一段合法寿司,以便自己能吃饱。
思路:
用num1和num2来记录一串连续的1和2的个数,用now1和now2来表示当前索引之前的区间是全1还是全2。
若在区间1里又读到1,则num1++;否则读到了2,此时区间可能为12或212,要更新答案为之前的12相等个数,然后清空开始的2的个数,让num2=1,并记录当前区间为2,即now2=1,now1=0。区间2同理。
总结:
一道模拟题,考察对连续区间的处理。
代码:
#include <iostream>
using namespace std;
int n;
int a[100010];
int ans=0; //有几个1,即ans为答案的一半
int num1=0,num2=0; //当前索引i之前的区间有多少1和2
bool now1=0,now2=0; //之前区间是1还是2
int main()
{
cin>>n;
for(int i=0;i<n;i++)
cin>>a[i];
if(a[0]==1) num1++,now1=1;
else if(a[0]==2) num2++,now2=1;
for(int i=1;i<n;i++)
{
if(a[i]==1)
{
if(now1) num1++;
if(now2) //到新区间,更新ans,清空另一个数
{
ans=max(ans,min(num1,num2));
num1=1;
now1=1,now2=0;
}
}
else if(a[i]==2)
{
if(now2) num2++;
if(now1)
{
ans=max(ans,min(num1,num2));
num2=1;
now1=0,now2=1;
}
}
}
//处理结尾的两个区间
ans=max(ans,min(num1,num2));
cout<<ans*2<<endl;
}