必做题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
问题分析
序列需要满足两点要求:1.连续 2.2的数量等于1的数量。
为了检验是否满足要求,定义变量sum1和sum2 统计1和2的数量。
为了保证是连续序列,我们从遍历开始,遇到不一样的数字就说明找到一个序列,再遇到不一样的数字说明这个序列结束,记录长度,更新sum1sum2.
因此定义变量t,初始化为0,由2变成1或者由1变成2的时候,让t++。当t等于2说明经历了两次变化,则当前序列结束,让t=1,更新sum1sum2。
代码出错的原因在于更新sum1sum2的时候出错,以[1,2,1,2,1,2]为例,在访问到a[3]=1的时候,应该讲sum1置为0。
代码实现
#include<stdio.h>
#include<algorithm>
using namespace std;
int n,a[100500];
int main()
{
scanf("%d",&n);
for(int i=1; i<=n; i++)
{
scanf("%d",&a[i]);
}
int suc=true;
for(int i=1; i<n; i++)
{
if(a[i] != a[i+1])
suc = false;
}
if(suc == true)
printf("0\n");
else{
int r=1;
int ans = 0;
int sum1=0,sum2=0;
int t=0;
while(r<=n)
{
if(r!=1 && a[r]!= a[r-1])
t++;
if(t==2)
{
ans = max(ans, min(sum1,sum2)*2);
t=1;
if(a[r-1] == 1)
sum2=0;//出错点
else
sum1=0;
}
if(a[r] == 1)
sum1++;
else
sum2++;
r++;
}
ans = max(ans, min(sum1,sum2)*2);
printf("%d\n",ans);
}
return 0;
}