“非常男女”计划题解

题目描述

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

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

输入格式

第一行有一个正整数 n (1≤n≤105)，代表学校的人数。

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

输出格式

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

如果不存在男女人数相等的子区间，请输出 0。

输入输出样例

输入 #1复制

9
0 1 0 0 0 1 1 0 0

输出 #1复制

6

思路

贪心即可。

代码见下

#include<bits/stdc++.h>
using namespace std;
long long n,a[100005],b[100005],c[200005][2],lk=0;
int main(){
	cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        b[i]=b[i-1]+a[i];
    }
    c[0][0]=0;
    for(int i=1;i<=n;i++){
        //cout<<b[i]*2-i<<endl;
        if(c[b[i]*2-i+100000][0]==0&&b[i]*2-i!=0){
            c[b[i]*2-i+100000][0]=i;
        }
        else{
            c[b[i]*2-i+100000][1]=i;
        }
    }
    for(int i=0;i<=200000;i++){
        lk=max(lk,c[i][1]-c[i][0]);
    }
    cout<<lk<<endl;
	return 0;
}