非常男女计划

【问题描述】

　　Matrix67已经当过多次“媒人”了。他因此获得了许多经验。例如，距Matrix67观察，身高相近的人似乎比较合得来。

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

【输入格式】

　　第一行有一个正整数n，代表学校的人数。
　　第二行有n个用空格隔开的数，这些数只能是0或1，其中，0代表一个女生，1代表一个男生。

【输出格式】

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

【输入样例】

9
0 1 0 0 0 1 1 0 0

【输出样例】

6

【数据范围】

对于30%的数据，n<=100；
对于50%的数据，n<=1 000；
对于100%的数据，n<=100 000。

【来源】

Matrix67原创

这道题我是用的前缀和的方法，d[1][i] 表示男生比女生多i个时是前几个人。d[0][i] 表示女生比男生多i个时是前几个人。当再找到相同情况是，直接减第一次出现这种情况的位置就可以了。所以可以在输入时就完成这道题。
时间复杂度O（n）（其实就是输入的复杂度）。
详细代码如下：

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=100005;

int n;
int d[2][maxn]={0},g[maxn]={0};

void init()
{
    int ans=0;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        g[i]=g[i-1];
        if(x==1) g[i]++;
        if(g[i]>i-g[i])
        {
            if(d[1][2*g[i]-i]==0)
            d[1][2*g[i]-i]=i;
            else ans=max(ans,i-d[1][2*g[i]-i]);
        }
        if(g[i]<i-g[i])
        {
            if(d[0][i-2*g[i]]==0)
            d[0][i-2*g[i]]=i;
            else ans=max(ans,i-d[0][i-2*g[i]]);
        }
        if(g[i]==i-g[i]) ans=max(ans,i);
    }
    printf("%d",ans);
}



int main()
{
    init();
    return 0;
}