[bzoj1593][Usaco2008 Feb]Hotel 旅馆 线段树

1593: [Usaco2008 Feb]Hotel 旅馆

Time Limit: 10 Sec   Memory Limit: 64 MB
[ Submit][ Status][ Discuss]

Description

奶牛们最近的旅游计划，是到苏必利尔湖畔，享受那里的湖光山色，以及明媚的阳光。作为整个旅游的策划者和负责人，贝茜选择在湖边的一家著名的旅馆住宿。这个巨大的旅馆一共有N (1 <= N <= 50,000)间客房，它们在同一层楼中顺次一字排开，在任何一个房间里，只需要拉开窗帘，就能见到波光粼粼的湖面。 贝茜一行，以及其他慕名而来的旅游者，都是一批批地来到旅馆的服务台，希望能订到D_i (1 <= D_i <= N)间连续的房间。服务台的接待工作也很简单：如果存在r满足编号为r..r+D_i-1的房间均空着，他就将这一批顾客安排到这些房间入住；如果没有满足条件的r，他会道歉说没有足够的空房间，请顾客们另找一家宾馆。如果有多个满足条件的r，服务员会选择其中最小的一个。 旅馆中的退房服务也是批量进行的。每一个退房请求由2个数字X_i、D_i 描述，表示编号为X_i..X_i+D_i-1 (1 <= X_i <= N-D_i+1)房间中的客人全部离开。退房前，请求退掉的房间中的一些，甚至是所有，可能本来就无人入住。 而你的工作，就是写一个程序，帮服务员为旅客安排房间。你的程序一共需要处理M (1 <= M < 50,000)个按输入次序到来的住店或退房的请求。第一个请求到来前，旅店中所有房间都是空闲的。

Input

* 第1行: 2个用空格隔开的整数：N、M

* 第2..M+1行: 第i+1描述了第i个请求，如果它是一个订房请求，则用2个数字 1、D_i描述，数字间用空格隔开；如果它是一个退房请求，用3 个以空格隔开的数字2、X_i、D_i描述

Output

* 第1..??行: 对于每个订房请求，输出1个独占1行的数字：如果请求能被满足 ，输出满足条件的最小的r；如果请求无法被满足，输出0

Sample Input

10 6
1 3
1 3
1 3
1 3
2 5 5
1 6

Sample Output

1
4
7
0
5

HINT

题目要求我们维护一棵01线段树

所以需要维护的值是区间最大0段，右端最多0的数量，左端最多0的数量

子树的端点个数随便搞都行

然后询问的时候优先级左大于中间大于右

说白了，就是一道01线段树模板题

早上刷水提神醒脑

#include<bits/stdc++.h>
using namespace std;
const int N = 50000 + 5;
int sum[N*4],lmx[N*4],rmx[N*4],siz[N*4],flag[N*4],n,m;
void update( int k ){
	sum[k] = max( max( sum[k<<1], sum[k<<1|1] ), lmx[k<<1|1]+rmx[k<<1] );
	lmx[k] = lmx[k<<1];
	if( sum[k<<1] == siz[k<<1] ) lmx[k] = sum[k<<1] + lmx[k<<1|1];
	rmx[k] = rmx[k<<1|1];
	if( sum[k<<1|1] == siz[k<<1|1] ) rmx[k] = sum[k<<1|1] + rmx[k<<1];
}
void pushdown( int k, int l, int  r ){
	if( flag[k] == 1 ){
		lmx[k<<1] = rmx[k<<1] = sum[k<<1] = siz[k<<1];
		lmx[k<<1|1] = rmx[k<<1|1] = sum[k<<1|1] = siz[k<<1|1];
		flag[k<<1] = flag[k<<1|1] = 1;
	}
	if( flag[k] == 0 ){
		lmx[k<<1] = rmx[k<<1] = sum[k<<1] = 0;
		lmx[k<<1|1] = rmx[k<<1|1] = sum[k<<1|1] = 0;
		flag[k<<1] = flag[k<<1|1] = 0;
	}
	flag[k] = -1;
}
void build( int k, int l, int r ){
	flag[k] = -1; lmx[k] = rmx[k] = sum[k] = siz[k] = r - l + 1;
	if( l == r ){ return; }
	int mid = l + r >> 1;
	build( k<<1, l, mid );
	build( k<<1|1, mid+1, r );
	update( k );
}
void change( int k, int l, int r, int L, int R, int delta ){
	if( l >= L && r <= R ){
		lmx[k] = rmx[k] = sum[k] = siz[k] * delta;
		flag[k] = delta;
		return;
	}
	pushdown( k, l, r );
	int mid = l + r >> 1;
	if( mid >= L ) change( k<<1, l, mid, L, R, delta );
	if( mid <  R ) change( k<<1|1, mid+1, r, L, R, delta );
	update( k );
}
int query( int k, int l, int r, int x ){
	pushdown( k, l, r );
	int mid = l + r >> 1;
	if( l == r ) return l;
	if( sum[k<<1] >= x ) return query( k<<1, l, mid, x );
	if( rmx[k<<1] + lmx[k<<1|1] >= x ) return mid - rmx[k<<1] + 1;
	return query( k<<1|1, mid+1, r, x );
}
int main(){
	scanf("%d%d", &n, &m);
	build( 1, 1, n );
	int opt, a, b;
	while( m-- ){
		scanf("%d%d", &opt, &a );
		if( opt == 1 ){
			if( sum[1] < a ){ puts("0"); continue; }
			int l = query( 1, 1, n, a );
			printf("%d\n", l );
			change( 1, 1, n, l, l+a-1, 0 );
		}else{
			scanf("%d", &b );
			change( 1, 1, n, a, a+b-1, 1 );
		}
	}
	return 0;
}
/*
10 6
1 3
1 3
1 3
1 3
2 5 5
1 6
*/