Description
Yzx已经当过多次“媒人”了。他因此获得了许多经验。例如,距Yzx观察,身高相近的人似乎比较合得来。
Yzx在学校策划了一次大型的“非常男女”配对活动。对于这次活动的参与者,Yzx有自己独特的选择方式。他希望能选择男女人数相等且身高都很接近的一些人。这种选择方式实现起来很简单,他让学校的所有人按照身高排成一排,然后从中选出连续的若干个人,使得这些人中男女人数相等。Yzx当然希望他能选出的人越多越好,请告诉他最多可以选出多少人来。
Input
第一行有一个正整数n,代表学校的人数。
第二行有n个用空格隔开的数,这些数只能是0或1,其中,0代表一个男生,1代表一个女生。
Output
一个非负整数,表示最长的一段男女人数相等的子序列长度(如果不存在男女人数相等的子序列输出0)。
Sample Input
9
0 1 0 0 0 1 1 0 0
Sample Output
6
Data Constraint
30%的数据,n<=100。
50%的数据,n<=1000。
100%的数据,n<=100000。
.
.
.
.
.
分析
前缀和+暴枚
枚举时,因为需要0与1配对,所以长度为+2
.
.
.
.
.
.
程序:
#include<iostream>
#include<cstdio>
#include<cmath>
using namespace std;
int sum[100010],n,a,ans=0;
int main()
{
scanf("%d",&n);
for (int i=1;i<=n;i++)
{
scanf("%d",&a);
sum[i]=sum[i-1]+a;
}
int i=1;
while (i<=n-ans)
{
int j=i+ans+1;
while (j<=n)
{
if ((sum[j]-sum[i-1])*2==j-i+1) ans=max(j-i+1,ans);
j+=2;
}
i++;
}
printf("%d",ans);
return 0;
}