Week11 D - 必做题11-4

Week11 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

解题思路
a1 记录当前有多少的1
a2 记录当前有多少的2
每次开始记录时将 a1 a2 归零
ans=max(ans,min(a1,a2))
Code

#include<iostream>
#include<algorithm>
using namespace std;
int n;
int a[100000];
int main(){
	cin>>n;
	for(int i=0;i<n;i++){
		cin>>a[i];
	}
	int s=0;
	int a1=0;
	int a2=0;
	int ans=0;
	for(int i=0;i<n;i++){
		if(a[i]==1){a1++;
		}else{a2++;
		}
		ans=max(ans,min(a1,a2));
		if(a[i+1]!=a[i]){
			if(a[i+1]==1){
				a1=0;
			}else{
				a2=0;
			}
		}
	}
	cout<<ans*2<<endl;
}