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
輸出
1
4
7
0
5
分析
以前都只做過線段樹的板子題,現如今碰到這道題,這感覺就跟沒學線段樹一樣……
不過還好,在gsj大佬的帶領下,還是搞出來了
訂房:查詢是否有連續D個位置是空的,如果有輸出最左端點
退房:將X到X+D-1這段位置置為空
這樣一轉述,就很容易想到線段樹了,區間查詢,區間修改
每個節點儲存3個資訊:
- 從左數最多有多少個連續的0 lmax
- 從右數最多有多少個連續的0 rmax
- 整個區間最多有多少個連續的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;
}