bzoj 1593: [Usaco2008 Feb]Hotel 旅馆 （线段树）

1593: [Usaco2008 Feb]Hotel 旅馆

Time Limit: 10 Sec   Memory Limit: 64 MB
Submit: 647   Solved: 361
[ 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

Source

Gold

[ Submit][ Status][ Discuss]

题解：线段树

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 1000000
using namespace std;
int ls[N*4],rs[N*4],tr[N*4],delta[N*4],n,m;
void build(int now,int l,int r)
{
	if (l==r) {
		tr[now]=ls[now]=rs[now]=1;
		return;
	}
	int mid=(l+r)/2;
	build(now<<1,l,mid);
	build(now<<1|1,mid+1,r);
	tr[now]=ls[now]=rs[now]=tr[now<<1]+tr[now<<1|1];
}
void pushdown(int now,int l,int r)
{
	if (!delta[now]) return;
	int mid=(l+r)/2;
	if (tr[now]) 
	 tr[now<<1]=ls[now<<1]=rs[now<<1]=mid-l+1,tr[now<<1|1]=ls[now<<1|1]=rs[now<<1|1]=r-mid;
	else 
	 tr[now<<1]=ls[now<<1]=rs[now<<1]=tr[now<<1|1]=ls[now<<1|1]=rs[now<<1|1]=0;
	delta[now<<1]=delta[now<<1|1]=1;
	delta[now]=0;
}
int find(int now,int l,int r,int x)
{
	if (l==r) return l;
	pushdown(now,l,r);
	int mid=(l+r)/2;
	if (tr[now<<1]>=x)  return find(now<<1,l,mid,x);
	else if (rs[now<<1]+ls[now<<1|1]>=x) return mid-rs[now<<1]+1;
	else return find(now<<1|1,mid+1,r,x);
}
void update(int now,int l,int r)
{
	tr[now]=max(tr[now<<1],max(tr[now<<1|1],rs[now<<1]+ls[now<<1|1]));
	int mid=(l+r)/2;
	ls[now]=ls[now<<1]+(ls[now<<1]==mid-l+1?ls[now<<1|1]:0);
	rs[now]=rs[now<<1|1]+(rs[now<<1|1]==r-mid?rs[now<<1]:0);
}
void qjchange(int now,int l,int r,int ll,int rr,int v)
{
	if (ll<=l&&r<=rr) {
		if (v) tr[now]=ls[now]=rs[now]=r-l+1;
		else tr[now]=ls[now]=rs[now]=0;
		delta[now]=1;
		return;
	}
	pushdown(now,l,r);
	int mid=(l+r)/2;
	if (ll<=mid) qjchange(now<<1,l,mid,ll,rr,v);
	if (rr>mid) qjchange(now<<1|1,mid+1,r,ll,rr,v);
	update(now,l,r);
}
int main()
{
   freopen("a.in","r",stdin);
   freopen("my.out","w",stdout);
   scanf("%d%d",&n,&m);
   build(1,1,n);
   for (int i=1;i<=m;i++) {
   	 int opt,x,y; scanf("%d",&opt);
   	 if (opt==1) {
   	 	 scanf("%d",&x);
   	 	 if (tr[1]<x) printf("0\n");
   	 	 else {
   	 	 	int t=find(1,1,n,x);
			printf("%d\n",t);
   	 	    qjchange(1,1,n,t,t+x-1,0);
		 }
		}
	  else {
	  	 scanf("%d%d",&x,&y);
	  	 qjchange(1,1,n,x,x+y-1,1);
	  }
   }	
}