洛谷P1114 “非常男女”计划

洛谷P1114 “非常男女”计划
map
O(n) 枚举过去
记录当前值最早出现的时间 x ，然后当这个值在 时，出现时间为 y 时
则 x+1--y 这段男女出现次数是为一样的，更新一下ans就行了

#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cstring>
#include <string>
#include <algorithm>
#include <iomanip>
#include <iostream>
using namespace std ; 

const int maxn = 100111 ; 
int n,sum,x,ans ; 
int a[maxn],b[maxn],map[maxn] ; 

inline int read() 
{
    char ch = getchar() ; 
    int x = 0,f = 1 ; 
    while(ch<'0'||ch>'9') { if(ch=='-') f = -1 ; ch = getchar() ; } 
    while(ch>='0'&&ch<='9') { x = x*10+ch-48 ; ch = getchar() ; }
    return x * f ; 
}

int main() 
{
    n = read() ; 
    for(int i=1;i<=n;i++) 
    {
        a[ i ] = read() ; 
        if(!a[i]) a[i] = -1 ; 
    }
    map[ 0 ] = 0 ; 
    for(int i=1;i<=n;i++ ) 
    {
        x+=a[ i ] ;
        if(!x) ans = max(ans,i) ; 
        if(map[ x ]) ans = max(ans,i - map[ x ] ) ;  
        if(!map[ x ]) map[ x ] = i ; 
    }
    printf("%d
",ans) ;  
    return 0 ; 
}