BZOJ 1637: [Usaco2007 Mar]Balanced Lineup( sort + 前缀和 )

时间:2022-06-12 06:29:57

BZOJ 1637: [Usaco2007 Mar]Balanced Lineup( sort + 前缀和 )

将 0 变为 -1 , 则只需找区间和为 0 , 即前缀和相同的最长区间 , 记录一下每个前缀和出现的最早和最晚的位置 , 比较一下就 OK 了

-------------------------------------------------------------------------------------------

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
 
#define rep( i , n ) for( int i = 0 ; i < n ; ++i )
#define clr( x , c ) memset( x , c , sizeof( x ) )
#define Rep( i , n ) for( int i = 1 ; i <=n ; ++i )
 
using namespace std;
 
const int maxn = 50000 + 5;
 
int n;
 
struct data {
int pos , kind , v;
bool operator < ( const data &rhs ) const {
return pos < rhs.pos;
}
};
 
data A[ maxn ];
int sum[ maxn ];
int L[ maxn ] , R[ maxn ];
 
int main() {
freopen( "test.in" , "r" , stdin );
cin >> n;
Rep( i , n ) {
data &x = A[ i ];
scanf( "%d%d" , &x.kind , &x.pos );
if( ! x.kind ) x.kind--;
}
sort( A + 1 , A + n + 1 );
sum[ 0 ] = 0;
Rep( i , n )
   sum[ i ] = sum[ i - 1 ] + A[ i ].kind;
clr( L , -1 );
clr( R , -1 );
Rep( i , n ) if( L[ sum[ i ] ] == -1 )
   L[ sum[ i ] ] = A[ i + 1 <= n ? i + 1 : i ].pos;
for( int i = n ; i > 0 ; i-- ) if( R[ sum[ i ] ] == -1 )
   R[ sum[ i ] ] = A[ i ].pos;
int ans = 0;
rep( i , maxn ) if( L[ i ] != -1 )
   ans = max( ans , R[ i ] - L[ i ] );
cout << ans << "\n";
return 0;
}

-------------------------------------------------------------------------------------------

1637: [Usaco2007 Mar]Balanced Lineup

Time Limit: 5 Sec  Memory Limit: 64 MB
Submit: 423  Solved: 283
[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
输入说明

有7头牛,像这样在数轴上。

1 1 0 1 0 1 1
+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
输出说明

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

<-------- 平衡的 -------->
1 1 0 1 0 1 1
+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+--+
0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25

HINT

Source