BZOJ 1593: [Usaco2008 Feb]Hotel 旅馆

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

题解

线段树。虽说无脑，但还是要细心。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<iostream>
#include<cstdlib>
#include<algorithm>
using namespace std;
int n,m;
struct shu{int l,r,s,lm,rm,sum,tag;} tr[200000];
//s:l到r的最长白色段；lm，rm：左右端的白色段长度；sum：段长度；tag：标记 
void build(int w,int b,int e)
{
	tr[w].l=b; tr[w].r=e; tr[w].tag=0;
	tr[w].lm=tr[w].rm=tr[w].sum=tr[w].s=e-b+1; 
	if(b==e) return;
	int mid=(b+e)>>1;
	build(w<<1,b,mid); build((w<<1)+1,mid+1,e);
}
void down(int w)
{
	int t=tr[w].tag; tr[w].tag=0;
	if(t==0) return;
	if(tr[w].l==tr[w].r) return;
	int z=w<<1,y=(w<<1)+1; 
	if(t==1)
	   {tr[z].lm=tr[z].rm=tr[z].s=0;
	    tr[y].lm=tr[y].rm=tr[y].s=0;
	   }
	else
	   {tr[z].lm=tr[z].sum; tr[z].rm=tr[z].sum; tr[z].s=tr[z].sum;
	    tr[y].lm=tr[y].sum; tr[y].rm=tr[y].sum; tr[y].s=tr[y].sum;
	   }
	tr[z].tag=tr[y].tag=t;
}
int find(int w,int x)
{
	down(w);
	int mid=(tr[w].l+tr[w].r)>>1;
	if(tr[w].l==tr[w].r) return tr[w].l;
	if(tr[w<<1].s>=x) return find(w<<1,x);
	else if(tr[w<<1].rm+tr[(w<<1)+1].lm>=x) return mid-tr[w<<1].rm+1;
	else return find((w<<1)+1,x);
}
void change(int w,int b,int e,int p)
{
	down(w);
	int l=tr[w].l,r=tr[w].r;
	if(l==b&&r==e)
	   {if(p==1) tr[w].lm=tr[w].rm=tr[w].s=0;
	    else
		   {tr[w].lm=tr[w].sum; tr[w].rm=tr[w].sum; tr[w].s=tr[w].sum;}
		tr[w].tag=p;
	    return;
	   }
	int mid=(l+r)>>1;
	if(mid>=e) change(w<<1,b,e,p);
	else if(mid<b) change((w<<1)+1,b,e,p);
	else {change(w<<1,b,mid,p); change((w<<1)+1,mid+1,e,p);}
	
	if(tr[w<<1].sum==tr[w<<1].s)
	   tr[w].lm=tr[w<<1].sum+tr[(w<<1)+1].lm;
	else  tr[w].lm=tr[w<<1].lm;
	if(tr[(w<<1)+1].sum==tr[(w<<1)+1].s)
	   tr[w].rm=tr[(w<<1)+1].sum+tr[w<<1].rm;
	else tr[w].rm=tr[(w<<1)+1].rm;
	tr[w].s=max(tr[w<<1].s,tr[(w<<1)+1].s);
	tr[w].s=max(tr[w].s,tr[w<<1].rm+tr[(w<<1)+1].lm);
}
int main()
{
	int i,x,y,z;
	scanf("%d%d",&n,&m);
	build(1,1,n);
	for(i=1;i<=m;i++)
	   {scanf("%d",&z);
	    if(z==1)
	       {scanf("%d",&x);
		    if(tr[1].s<x) printf("0\n");
		    else
			   {y=find(1,x); printf("%d\n",y);
			    change(1,y,y+x-1,1);
			   }
		   }
		else
		   {scanf("%d%d",&x,&y);
		    change(1,x,x+y-1,2);
		   }
	   }
	return 0;
}