题目描述
近来,初一年的XXX小朋友致力于研究班上同学的配对问题(别想太多,仅是舞伴),通过各种推理和实验,他掌握了大量的实战经验。例如,据他观察,身高相近的人似乎比较合得来。
万圣节来临之际,XXX准备在学校策划一次大型的“非常男女”配对活动。对于这次活动的参与者,XXX有自己独特的选择方式。他希望能选择男女人数相等且身高都很接近的一些人。这种选择方式实现起来很简单。他让学校的所有人按照身高排成一排,然后从中选出连续的若干个人,使得这些人中男女人数相等。为了使活动更热闹,XXX当然希望他能选出的人越多越好。请编写程序告诉他,他最多可以选出多少人来。
输入输出格式
输入格式:
第一行有一个正整数n,代表学校的人数。n<=100000
第二行有n个用空格隔开的数,这些数只能是0或1,其中,0代表一个女生,1代表一个男生
输出格式:
输出一个非负整数。这个数表示在输入数据中最长的一段男女人数相等的子序列长度。
如果不存在男女人数相等的子序列,请输出0。
输入输出样例
输入样例#1:
9 0 1 0 0 0 1 1 0 0
输出样例#1:
6
思路:
利用相对差,Diff[i]数组表示i位置男生人数-女生人数的差值。
代码:
/*注释部只有70分,哪错了但我找不出来。。*/
1 /* Wrong:70 */ 2 #include<cstdio> 3 #include<iostream> 4 using namespace std; 5 6 int n,cnt,Ans,Sum[2],Record[100005],Diff[200005],From[200005],To[200005]; 7 bool vis[200005]; 8 9 int main() 10 { 11 scanf("%d",&n); 12 for(int i=1;i<=n;i++) 13 { 14 int a; 15 scanf("%d",&a); 16 //if(a) ++Sum[1]; 17 //else ++Sum[0]; 18 Diff[i]=Diff[i-1]; 19 if(a) 20 ++Diff[i]; 21 else 22 --Diff[i]; 23 /*if(!vis[Diff[i]+n]) 24 vis[Diff[i]+n]=1,Record[++cnt]=Diff[i]+n,From[Diff[i]+n]=i; 25 else 26 To[Diff[i]+n]=i;*/ 27 } 28 /*for(int i=1;i<=n;i++) 29 printf("%d :%d ",i,Diff[i]);*/ 30 /*for(int i=1;i<=n;i++) 31 Ans= Ans<(To[Diff[i]+n]-From[Diff[i]+n])?(To[Diff[i]+n]-From[Diff[i]+n]):Ans;*/ 32 for(int i=1;i<=n;i++) 33 { 34 if(!Diff[i]) Ans= Ans<i?i:Ans;//此时男女相等 35 else if(To[Diff[i]+n]) Ans=max(Ans,i-To[Diff[i]+n]); 36 else To[Diff[i]+n]=i; 37 } 38 printf("%d",Ans); 39 return 0; 40 }