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
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;
}