bzoj 1637: [Usaco2007 Mar]Balanced Lineup

1637: [Usaco2007 Mar]Balanced Lineup

Time Limit: 5 Sec   Memory Limit: 64 MB
Submit: 689   Solved: 456
[ Submit][ Status][ Discuss]

Description

Farmer John 决定给他的奶牛们照一张合影，他让 N (1 ≤ N ≤ 50,000) 头奶牛站成一条直线，每头牛都有它的

坐标(范围: 0..1,000,000,000)和种族(0或1)。 一直以来 Farmer John 总是喜欢做一些非凡的事，当然这次照相

也不例外。他只给一部分牛照相，并且这一组牛的阵容必须是“平衡的”。平衡的阵容，指的是在一组牛中，种族0和种族1的牛的数量相等。 请算出最广阔的区间，使这个区间内的牛阵容平衡。区间的大小为区间内最右边的牛的坐标减去最做边的牛的坐标。 输入中，每个种族至少有一头牛，没有两头牛的坐标相同。

Input

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

Output

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

Sample Input

7

0 11

1 10

1 25

1 12

1 4

0 13

1 22

Sample Output

11

经典题

记录前缀和，如果牛的种类是1：sum++，如果牛的种类是0：sum--

如果在某个过程sum=0说明此时两个种类牛的数量相等

但是这样只能得出种类数量相同的最长前缀，而题目中要求可以是中间某一段

这样对于sum开个map统计一下就好，对于当前sum查一下最早得出sum是在什么位置，那中间那一段也满足要求

#include<stdio.h>
#include<map>
#include<algorithm>
using namespace std;
typedef struct Point
{
	int loc;
	int x;
	bool operator < (const Point &b) const
	{
		if(loc<b.loc)
			return 1;
		return 0;
	}
}Point;
Point s[50005];
map<int, int> p;
int main(void)
{
	int n, i, sum, ans;
	scanf("%d", &n);
	for(i=1;i<=n;i++)
	{
		scanf("%d%d", &s[i].x, &s[i].loc);
		if(s[i].x==0)
			s[i].x = -1;
	}
	sort(s+1, s+n+1);
	ans = sum = 0;
	p[0] = s[1].x;
	for(i=1;i<=n;i++)
	{
		sum += s[i].x;
		if(p.count(sum))
			ans = max(ans, s[i].loc-p[sum]);
		if(p.count(sum)==0)
			p[sum] = s[i+1].loc;
	}
	printf("%d\n", ans);
	return 0;
}