BZOJ 1637: [Usaco2007 Mar]Balanced Lineup【hash】

1637: [Usaco2007 Mar]Balanced Lineup

【题目描述】
传送门

【题解】

我们可以用前缀和的想法，如果是0那么$Sum[i]=Sum[i-1]-1$，否则$Sum[i]=Sum[i-1]+1$，如果$Sum[i]==Sum[j](i<j)$那么$i+1$到$j$中的两种牛的个数是一样的，所以可以用一个数组记住$Sum[i]$的位置，然后刷一个最大值就可以了。

代码如下

#include<cstdio>
#include<algorithm>
using namespace std;
int n,Sum[50005],Ans;
struct HASH{
    int a[2*50005];
    int &operator [](const int b){return a[b];}
}hsh;
struct A{
    int id,x;
    bool operator <(const A b)const{return x<b.x;}
}a[50005];
int main(){
    #ifndef ONLINE_JUDGE
    freopen("prob.in","r",stdin);
    freopen("prob.out","w",stdout);
    #endif
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d%d",&a[i].id,&a[i].x);
    sort(a+1,a+1+n);
    for(int i=1;i<=n;i++){
        Sum[i]=Sum[i-1]+(a[i].id?1:-1);
        if(!hsh[Sum[i]]) hsh[Sum[i]]=i;
        else Ans=max(Ans,a[i].x-a[hsh[Sum[i]]+1].x);
    }
    printf("%d\n",Ans);
    return 0;
}