近来,蒜头君致力于研究班上同学的配对问题(别想太多,仅是舞伴),通过各种推理和实验,他掌握了大量的实战经验。例如,据他观察,身高相近的人似乎比较合得来。
万圣节来临之际,蒜头君准备在学校策划一次大型的“非常男女”配对活动。对于这次活动的参与者,蒜头君有自己独特的选择方式。他希望能选择男女人数相等且身高都很接近的一些人。这种选择方式实现起来很简单。他让学校的所有人按照身高排成一排,然后从中选出连续的若干个人,使得这些人中男女人数相等。为了使活动更热闹,蒜头君当然希望他能选出的人越多越好。请编写程序告诉他,他最多可以选出多少人来。
输入格式
第一行有一个正整数 n ( 1 ≤ n ≤ 10^5 ),代表学校的人数。
第二行有 n 个用空格隔开的数,这些数只能是 0 或 1,其中,0 代表一个女生,1代表一个男生。
输出格式
输出一个非负整数,这个数表示在输入数据中最长的一段男女人数相等的连续子序列长度。
#include<bits/stdc++.h>
using namespace std;
int N;
int a[100005];
int main()
{
cin >> N;
for(int i=1; i<=N; i++)
{
cin >> a[i];
if(a[i] == 0) a[i] = -1;//男为1,女为-1,男女相等和为0。
a[i] += a[i-1];//构建前缀和数组。
}
int ans = 0;
for(int i=0; i<=N; i++)
{
for(int j=N; j>=i+1; j--)//从后向前遍历,更快找到最大人数.
{
if(a[i] == a[j])//第i个和第j个为相等,则i+1到j这j-i个人男女比例相等。
{
ans = max(ans, j-i);
break;
}
}
}
cout << ans << endl;
return 0;
}