1. 程式人生 > >1016 - 線段樹 - 旅館(BZOJ xxx)

1016 - 線段樹 - 旅館(BZOJ xxx)

題目描述

奶牛們最近的旅遊計劃,是到蘇必利爾湖畔,享受那裡的湖光山色,以及明媚的陽光。作為整個旅遊的策劃者和負責人,貝茜選擇在湖邊的一家著名的旅館住宿。這個巨大的旅館一共有 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)個按輸入次序到來的住店或退房的請求。第一個請求到來前,旅店中所有房間都是空閒的。

輸入格式

第 1 行: 2 個用空格隔開的整數:N、M 。
第 2.. M+1行: 第 i+1 描述了第 i 個請求,如果它是一個訂房請求,則用 2 個數字 1、D_i描述,數字間用空格隔開;如果它是一個退房請求,用 3 個以空格隔開的數字 2、X_i、D_i 描述.

輸出格式

第 1 ... ?? 行: 對於每個訂房請求,輸出 1 個獨佔 1 行的數字:如果請求能被滿足,輸出滿足條件的最小的 r;如果請求無法被滿足,輸出 0 。

樣例資料 1

輸入 

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

輸出





5

 

分析

以前都只做過線段樹的板子題,現如今碰到這道題,這感覺就跟沒學線段樹一樣……

不過還好,在gsj大佬的帶領下,還是搞出來了

訂房:查詢是否有連續D個位置是空的,如果有輸出最左端點

退房:將X到X+D-1這段位置置為空

這樣一轉述,就很容易想到線段樹了,區間查詢,區間修改

每個節點儲存3個資訊:

  1. 從左數最多有多少個連續的0  lmax
  2. 從右數最多有多少個連續的0  rmax
  3. 整個區間最多有多少個連續的0  maxn

 一開始的時候這些的值都賦為區間的長度,因為整個區間都是空的

當遇到一個查詢的時候,就從root開始,如果左邊的連續0的個數大於我們需要的D,就直接返回 左端點

然後判斷中間

最後判斷右邊

(因為題目要求我們輸出最左端的一種情況,每次我們都從左開始判斷,就能保證最後得到的答案一定是最靠左的了)

遇到修改的時候,就該怎麼修改怎麼修改,只是在最後pushup的時候稍微複雜一點點

有幾種情況需要判斷

最後需要注意:由於我們涉及了區間修改,所以每做一步操作都別忘了pushdown

 

程式碼

#include<bits/stdc++.h>
#define in read()
#define lc (k<<1)
#define rc (k<<1)|1//
#define N 50009
using namespace std;
inline int read(){
	char ch;int f=1,res=0;
	while((ch=getchar())<'0'||ch>'9') if(ch=='-') f=-1;
	while(ch>='0'&&ch<='9'){
		res=(res<<3)+(res<<1)+ch-'0';
		ch=getchar();
	}
	return f==1?res:-res;
}
int n,m;
int lmax[4*N],rmax[4*N],maxn[4*N],lzy[4*N];
void build(int k,int l,int r){
	if(l==r){
		lmax[k]=rmax[k]=maxn[k]=1;
		return ;
	}
	int mid=l+r>>1;
	build(lc,l,mid);build(rc,mid+1,r);
	lmax[k]=rmax[k]=maxn[k]=r-l+1;
}
void pushdown(int k,int l,int r){
	int mid=l+r>>1;
	if(!maxn[k]){
		lmax[lc]=rmax[lc]=maxn[lc]=0;
		lmax[rc]=rmax[rc]=maxn[rc]=0;
	}
	else{
		lmax[lc]=rmax[lc]=maxn[lc]=mid-l+1;
		lmax[rc]=rmax[rc]=maxn[rc]=r-mid;
	}
	lzy[k]=0;lzy[lc]=lzy[rc]=1;
}
int query(int k,int l,int r,int x){
	if(l==r) return l;
	int mid=l+r>>1;
	if(lzy[k]) pushdown(k,l,r);
	if(lmax[k]>=x) return l;
	if(maxn[lc]>=x) return query(lc,l,mid,x);
	if(rmax[lc]+lmax[rc]>=x) return mid-rmax[lc]+1;
	return query(rc,mid+1,r,x);
}
void pushup(int k,int l,int r){
	int mid=l+r>>1;
	if(lmax[lc]!=mid-l+1) lmax[k]=lmax[lc];
	else lmax[k]=lmax[lc]+lmax[rc];
	if(rmax[rc]!=r-mid) rmax[k]=rmax[rc];
	else rmax[k]=rmax[lc]+rmax[rc];
	maxn[k]=max(lmax[k],rmax[k]);
	maxn[k]=max(maxn[k],rmax[lc]+lmax[rc]);
	maxn[k]=max(maxn[k],maxn[lc]);maxn[k]=max(maxn[k],maxn[rc]);
}
void modify(int k,int l,int r,int x,int y,int v){
	if(x<=l&&r<=y){
		if(!v){
			lmax[k]=rmax[k]=maxn[k]=r-l+1;
			lzy[k]=1;
		}
		else{
			lmax[k]=rmax[k]=maxn[k]=0;
			lzy[k]=1;
		}
		return;
	}
	if(lzy[k]) pushdown(k,l,r);
	int mid=l+r>>1;
	if(x<=mid) modify(lc,l,mid,x,y,v);
	if(y>mid) modify(rc,mid+1,r,x,y,v);
	pushup(k,l,r);
}
int main(){
	n=in;m=in;
	int i,j,k,ty,x,d;
	build(1,1,n);
	for(i=1;i<=m;++i){
		ty=in;
		if(ty==1){
			d=in;
			if(maxn[1]<d) printf("0\n");
			else{
				int pos=query(1,1,n,d);
				printf("%d\n",pos);
				modify(1,1,n,pos,pos+d-1,1);//1-->be used
			}
		}
		else{
			x=in;d=in;
			modify(1,1,n,x,x+d-1,0);
		}
	}
	return 0;
}