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
输出:一个整数(代表东东可以选择的最长的一段连续的且合法的寿司)
Sample Input 1
7
2 2 2 1 1 2 2
Sample Output 1
4
Sample Input 2
6
1 2 1 2 1 2
Sample Output 2
2
Sample Input 3
9
2 2 1 1 1 2 2 2 2```
### Sample Output 3
```handlebars
6
题解
遍历所有字母,对于第i个寿司A,我们需要记录到本寿司为止,连续的A寿司数量有多少个了,在这一段A寿司之前有多少个连续的B寿司。取这两部分的最小值乘2,就是到目前为止东东能选取的最大寿司数量。
用f2记录到当前字母为止连续的A寿司数量,f1记录这一段A类寿司之前的连续的B类寿司的数量。初始化f1=0,f2=1,last为输入的第一个,most(记录最长可选寿司数量)初始化为0,从第二个开始依次遍历每个寿司,如果新输入的寿司与上一个相同,f2++;如果新输入的寿司与上一个不同,则说明上一段寿司该截止了,更新most为max(most,min(2×f1,2×f2)),令f2=1,f1=f2,last为新输入的寿司类型。这样循环到结束,再更新一次most(因为最后一段的寿司数量没有被计算过)。
最后输出most即为答案。
代码
#include <iostream>
#include <stdio.h>
#include <vector>
using namespace std;
int main(int argc, char** argv) {
int n;scanf("%d",&n);
int f1=0,f2=0;
int last;
int most=0;
scanf("%d",&last);f2=1;
for(int i=2;i<=n;i++){
int now;scanf("%d",&now);
if(now==last){
f2++;
}else{
if(f2>f1){
if(2*f1>most){most=2*f1;}
}else{
if(2*f2>most){most=2*f2;}
}
f1=f2;f2=1;
last=now;
}
}
if(f2>f1){
if(2*f1>most){most=2*f1;}
}else{
if(2*f2>most){most=2*f2;}
}
cout<<most<<endl;
return 0;
}