luoguP1114 “非常男女”计划 [前缀和]

题目描述

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

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

输入格式

第一行有一个正整数nn，代表学校的人数。n \le 100000n≤100000

第二行有nn个用空格隔开的数，这些数只能是00或11，其中，00代表一个女生，11代表一个男生

输出格式

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

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

输入输出样例

输入 #1复制

9
0 1 0 0 0 1 1 0 0

输出 #1复制

6

Solution
虽然这题本身不是太难。。不过利用前缀和的方法可以说比较巧妙（可能是我太久没做了，现在太蠢了）
处理到第i项，考虑记录sum[0/1]来表示前i项中男生和女生的总和，记sum[0]-sum[1]=x，若前l项和前r项的x值相等(l<=r)，则l-r合法，对于每个x值，保证l最小即r最大即可

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

const int maxn=100005;

int n,ans,GG,MM;
int ll[maxn<<1],rr[maxn<<1];

int main(){
//    freopen("temp.in","r",stdin);
    
    scanf("%d",&n);
    for(int i=1,tmp,t;i<=n;i++){
        scanf("%d",&tmp);
        if(tmp)  GG++;
        else  MM++;
        t=MM-GG+n;
        if(!ll[t]&&t!=n)  ll[t]=i;
        else  rr[t]=i;
    }
    for(int i=0;i<=2*n;i++)  ans=max(ans,rr[i]-ll[i]);
    printf("%d\n",ans);
    
    return 0;
}