首页 > 代码库 > 洛谷 P1114 “非常男女”计划

洛谷 P1114 “非常男女”计划

题目描述

近来,初一年的XXX小朋友致力于研究班上同学的配对问题(别想太多,仅是舞伴),通过各种推理和实验,他掌握了大量的实战经验。例如,据他观察,身高相近的人似乎比较合得来。

万圣节来临之际,XXX准备在学校策划一次大型的“非常男女”配对活动。对于这次活动的参与者,XXX有自己独特的选择方式。他希望能选择男女人数相等且身高都很接近的一些人。这种选择方式实现起来很简单。他让学校的所有人按照身高排成一排,然后从中选出连续的若干个人,使得这些人中男女人数相等。为了使活动更热闹,XXX当然希望他能选出的人越多越好。请编写程序告诉他,他最多可以选出多少人来。

输入输出格式

输入格式:

第一行有一个正整数n,代表学校的人数。n<=100000

第二行有n个用空格隔开的数,这些数只能是0或1,其中,0代表一个女生,1代表一个男生

输出格式:

输出一个非负整数。这个数表示在输入数据中最长的一段男女人数相等的子序列长度。

如果不存在男女人数相等的子序列,请输出0。

/*还是那天的前缀和思想,用 一个记录前缀和的数组,讲0
看做-1记录。遍历时,找到与当前前缀和相等的最早的数字,
(用另一个数组记录)之后 位置相减即可*/
#include<cstdio>
#include<algorithm>
#define maxn 100001
#define INF 100000
using namespace std;
int sum[maxn];
int map[maxn*2];
int n,maxx=-1;
int read()
{
int now=0;
char c=getchar();
while(c>‘9‘||c<‘0‘)c=getchar();
while(c>=‘0‘&&c<=‘9‘)
{
now*=10;
now+=c-‘0‘;
c=getchar();
}
return now;
}
int main()
{
n=read();
for(int i=1;i<=n;i++)
{
int num=read();
num==1?num=1:num=-1;
sum[i]+=num+sum[i-1];
}
for(int i=0;i<=200001;i++)map[i]=-1;
for(int i=0;i<=n;i++)
{
if(map[sum[i]+INF]==-1)
{
map[sum[i]+INF]=i;
}
if(map[sum[i]+INF]!=-1)
maxx=max(maxx,i-map[sum[i]+INF]);
}
printf("%d",maxx);
return 0;
}

洛谷 P1114 “非常男女”计划