洛谷 P1114 “非常男女”计划

Posted 2020-09-19 SovietPower

To 洛谷.1114 “非常男女”计划

题目描述

近来，初一年的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\n",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 }