BZOJ 1637: [Usaco2007 Mar]Balanced Lineup

Description

Farmer John 决定给他的奶牛们照一张合影，他让 N (1 ≤ N ≤ 50,000) 头奶牛站成一条直线，每头牛都有它的坐标(范围: 0..1,000,000,000)和种族(0或1)。 一直以来 Farmer John 总是喜欢做一些非凡的事，当然这次照相也不例外。他只给一部分牛照相，并且这一组牛的阵容必须是“平衡的”。平衡的阵容，指的是在一组牛中，种族0和种族1的牛的数量相等。 请算出最广阔的区间，使这个区间内的牛阵容平衡。区间的大小为区间内最右边的牛的坐标减去最做边的牛的坐标。 输入中，每个种族至少有一头牛，没有两头牛的坐标相同。

Input

第一行: 一个整数: N 。

第2..N + 1行: 每行两个整数，为种族 ID 和 x 坐标。

Output

第一行: 一个整数，阵容平衡的最大的区间的大小。

Sample Input

7
0 11
1 10
1 25
1 12
1 4
0 13
1 22

Sample Output

11
输入说明
有7头牛，像这样在数轴上。 

              1                               1   0   1  0                                         1           1
+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
0  1   2   3  4   5  6   7   8  9 1011121314 1516 171819 2021222324 25
输出说明

牛 #1 (at 11), #4 (at 12), #6 (at 13), #7 (at 22) 组成一个平衡的最大的区间，大小为 22-11=11 个单位长度。

题解

开始想dp，但感觉都是n^2.后来想想如果把0换成-1，再用前缀和s[],找到最长的s[i]==s[j]（i>j）即可。

#include<cstdio> 
#include<cstring> 
#include<iostream> 
#include<algorithm> 
#include<cstdlib> 
#include<cmath> 
#define cc 50000 
using namespace std; 
int n,s[50002],pd[100002],ans; 
struct shu {int v,x;} a[50002]; 
bool kp(const shu &i,const shu &j) {return i.x<j.x;} 
void init() 
{ 
    scanf("%d",&n); 
    int i; 
    for(i=1;i<=n;i++) 
       {scanf("%d%d",&a[i].v,&a[i].x); 
        if(a[i].v==0) a[i].v=-1; 
       } 
    sort(a+1,a+n+1,kp); 
} 
void work() 
{ 
    int i,j; 
    memset(pd,-1,sizeof(pd)); 
    for(i=1;i<=n;i++) 
       {s[i]=s[i-1]+a[i].v; 
        j=s[i]+cc; 
        if(pd[j]==-1) pd[j]=a[i+1].x;//这里要注意 
        else ans=max(a[i].x-pd[j],ans); 
       } 
} 
int main() 
{ 
    init(); work(); 
    printf("%d\n",ans); 
    return 0; 
}