bzoj1637[Usaco2007 Mar]Balanced Lineup*

bzoj1637[Usaco2007 Mar]Balanced Lineup

题意：

n头牛，第i头牛位置为ai，种族为bi（只能为0，1），求一个区间（按数轴位置），使得区间两端牛距离差最大且两种种族牛数相等。n≤50000。

题解：

按位置排序。然后利用前缀和sum[i][0]-sum[j-1][0]=sum[i][1]-sum[j-1][1]，sum[i][0]-sum[i][1]=sum[j-1][0]-sum[j-1][1]，故用个set维护sum[i][0]-sum[i][1]的值，每次找一下set里有没有与当前两个sum元素差相等的值，距离差和答案比较一下。

代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <set>
#define inc(i,j,k) for(int i=j;i<=k;i++)
#define maxn 50010
#define INF 0x3fffffff
using namespace std;

inline int read(){
    char ch=getchar(); int f=1,x=0;
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
    while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
    return f*x;
}
struct nd{int id,v; bool operator < (const nd &a)const{return v<a.v;};};
nd nds[maxn]; set<nd>st; int n,sm[2][maxn],ans;
int main(){
    n=read(); inc(i,1,n){int x=read(),y=read(); nds[i]=(nd){x,y};} sort(nds+1,nds+1+n);
    inc(i,1,n)sm[0][i]=sm[0][i-1],sm[1][i]=sm[1][i-1],sm[nds[i].id][i]++;
    inc(i,1,n){
        set<nd>::iterator sti=st.find((nd){0,sm[0][i]-sm[1][i]});
        if(sti!=st.end()){ans=max(ans,nds[i].v-nds[sti->id+1].v);}
        else st.insert((nd){i,sm[0][i]-sm[1][i]});
    }
    printf("%d",ans); return 0;
}

 

20160922