POJ 3667 Hotel 线段树

题意：  奶牛们最近的旅游计划，是到苏必利尔湖畔，享受那里的湖光山色，以明媚的阳光。作为整个旅游的策划者和负责人，贝茜选择在湖边的一家著名的旅馆住宿。这个巨大的旅馆一共有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)个按输入次序到来的住店或退房的请求。第一个请求到来前，旅店中所有房间都是空闲的。

题解：节点中 lsum表示从该节点的最左端起的连续房间数， rsum 表示该节点最右端的连续房间数， sum 表示该节点内最大的连续房间数。flag == 1 表示房间全空可以入住， flag == 0 表示客满不能入住。flag == -1 表示不用向下更新。

#include<iostream>
using namespace std;

#define N 50003
#define L(u) ( u << 1 )
#define R(u) ( u << 1 | 1 )
#define max(a,b) ( a > b ? a : b )

struct TreeNode
{
	int l, r, flag;
	int lsum, rsum, msum;
	void cal ()
	{
		lsum = rsum = msum = (r - l + 1) * flag;
	}
} node[N*3];

void build ( int u, int l, int r )
{
	node[u].r = r;
	node[u].l = l;
	node[u].flag = 1;
	node[u].cal();
	if ( l == r ) return;
	int mid = ( l + r ) >> 1;
	build ( L(u), l, mid );
	build ( R(u), mid+1, r );
}

void update ( int u, int l, int r, int flag )
{
	if ( l <= node[u].l && node[u].r <= r )
	{
		node[u].flag = flag;
		node[u].cal();
		return;
	}

	if ( node[u].flag != -1 )
	{
		node[L(u)].flag = node[R(u)].flag = node[u].flag;
		node[L(u)].cal();
		node[R(u)].cal();
		node[u].flag = -1;
	}

	int mid = ( node[u].l + node[u].r ) >> 1;
	if ( r <= mid )
		update ( L(u), l, r, flag );
	else if ( l > mid )
		update ( R(u), l, r, flag );
	else
	{
		update ( L(u), l, mid, flag );
		update ( R(u), mid + 1, r, flag );
	}

	if ( node[L(u)].lsum == node[L(u)].r - node[L(u)].l + 1 )
		node[u].lsum = node[L(u)].lsum + node[R(u)].lsum;
	else node[u].lsum = node[L(u)].lsum;

	if ( node[R(u)].rsum == node[R(u)].r - node[R(u)].l + 1 )
		node[u].rsum = node[L(u)].rsum + node[R(u)].rsum;
	else node[u].rsum = node[R(u)].rsum;
          //这句话是错误的node[u].msum = max ( max(node[u].lsum, node[u].rsum), node[L(u)].rsum+node[R(u)].lsum );
	node[u].msum = max ( max(node[L(u)].msum, node[R(u)].msum ), node[L(u)].rsum+node[R(u)].lsum );
}

int query ( int u, int num )
{
	if ( node[u].l == node[u].r && num == 1 )
		return node[u].l;

	if ( node[u].flag != -1 )
	{
		node[L(u)].flag = node[R(u)].flag = node[u].flag;
		node[L(u)].cal();
		node[R(u)].cal();
		node[u].flag = -1;
	}

	if ( node[L(u)].msum >= num )
		return query ( L(u), num );
	else if ( node[L(u)].rsum + node[R(u)].lsum >= num )
		return node[L(u)].r - node[L(u)].rsum + 1;
	else if ( node[R(u)].msum >= num )
		return query ( R(u), num );
	return 0;
}

int main()
{
	int n, m, oper, s, num;    
	while ( scanf("%d%d",&n,&m) != EOF )
	{
		build ( 1, 1, n );
		while ( m-- )
		{ 
			scanf("%d",&oper);
			if ( oper == 1 )
			{
				scanf("%d",&num);
				s = query ( 1, num );
				printf("%d\n",s);
				if ( s != 0 )
					update ( 1, s, s+num-1, 0 );
			}
			else
			{
				scanf("%d%d",&s,&num);
				update ( 1, s, s+num-1, 1 );
			}
		}
	}
	return 0;
}