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
输出:一个整数(代表东东可以选择的最长的一段连续的且合法的寿司)
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
二、思路与算法
一开始使用尺取的方法,每次先找到左指针位置,再去找右指针位置,提交测试下来,前27个数据都确实没有问题,但是第28个数据总是超时。
在努力剪枝、简化之后依旧超时,最后确定是大体上算法就有问题。
分析之后确实,while多重嵌套,参考网络上的做法之后,决定改为类似于【打表】的方法,只遍历整个寿司序列一次,val数组记录所有相邻相同数字的数目(表述得有点不准确,eg寿司序列是2211222,那么val记录的就是223),最后求出val相邻元素最小值的最大值,就是最终答案。
三、代码实现
#include<cstdio>
#include<algorithm>
using namespace std;
int n=0; //序列长度
int shou[100010]={0};
int left=1,right=1; //左右指针
int num=0,ans=0;
int val[100010]={0};
int cnt=0;
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d",&shou[i]);
}
while(left<=n){
right=left;
while(shou[right]==shou[left]&&right<=n){right++; }//right是第一个变的位置
num=right-left;
val[cnt++]=num;
left=right;
}
for(int i=0;i<cnt-1;i++){
int temp=min(val[i],val[i+1]);
ans=max(ans,temp+temp);
}
printf("%d\n",ans);
return 0;
}