“非常男女”计划-CSDN博客

题目描述
近来，初一年的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
.
.
.
.
.
.
.

分析

采用赋值法，将男赋值为1，女为-1，记录一下前缀和
我们可以从1到n扫过去，看看前面有没有和我们相同的前缀和（这样减掉的话男女数量就相等了），我用map【i】数组记录i这个前缀和出现的最早位置
注意：由于c++数组下标不能为负，所以都加上n；前缀和为0特殊处理，因为我那个map【i】=0的时候是当没出现过的，不然map[0]=0也会当没出现过。
.
.
.
.
.
.
.

程序：

#include<iostream>
#include<cstdio>
using namespace std;
int a[100001],Map[200001],sum[200001],n,i,ans;
int main()
{
    scanf("%d",&n);
    for (i=1;i<=n;i++)
    {
        scanf("%d",&a[i]); 
        if (a[i]==0) a[i]=-1; 
        sum[i]=sum[i-1]+a[i];
    }
    for (i=1;i<=n;i++)
    {
        if (sum[i]==0)
        {
            ans=max(ans,i);
            continue;
        }
        if (Map[sum[i]+n]&&ans<i-Map[sum[i]+n]) ans=i-Map[sum[i]+n];
        if (Map[sum[i]+n]==0) Map[sum[i]+n]=i;
    }
    cout<<ans<<endl;
}