P1114 “非常男女”计划

点击查看代码

#include<bits/stdc++.h>
using namespace std;

const int N=1e5+10,OFFSET=1e5;
int n;
int pos[2*N+100];

int main()
{
    ios::sync_with_stdio(0),cin.tie(0);

    cin>>n;
    memset(pos,-1,sizeof pos);
    pos[0+OFFSET]=0;

    int ans=0;
    int cur=0;
    for(int i=1;i<=n;i++){
        int x;
        cin>>x;
        if(x==1) cur++;
        else cur--;

        
            int var=cur+OFFSET;
            if(pos[var]!=-1){
                ans=max(ans,i-pos[var]);
            }else{
                pos[var]=i;
            }
        }
    

    cout<<ans<<endl;

    return 0;
}