BZOJ 1593: [Usaco2008 Feb]Hotel 旅馆

时间:2021-08-21 03:05:29

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

题解:

在线段树上二分第一个>=d的位置。

代码:

#include<cstdio>
#include<algorithm>
#include<cstring>
//by zrt
//problem:
using namespace std;
int n,m,opt,x;
const int MAXN=50005,MAXM=50005;
int lft[MAXN*3],rgt[MAXN*3],sum[MAXN*3],_l,_r;
bool empty[MAXN*3],set[MAXN*3];
void mk(int o,int l,int r){
if(l==r){
lft[o]=rgt[o]=sum[o]=1;
}else{
int mid=(l+r)>>1;
mk(o<<1,l,mid);
mk(o<<1|1,mid+1,r);
lft[o]=rgt[o]=sum[o]=sum[o<<1]+sum[o<<1|1];
}
}
void pushdown(int o,int l,int r){
int mid=(l+r)>>1;
if(empty[o]){
lft[o<<1]=rgt[o<<1]=sum[o<<1]=mid-l+1;
lft[o<<1|1]=rgt[o<<1|1]=sum[o<<1|1]=r-mid;
empty[o<<1]=empty[o<<1|1]=1;set[o<<1]=set[o<<1|1]=empty[o]=0;
}
if(set[o]){
lft[o<<1]=rgt[o<<1]=sum[o<<1]=0;
lft[o<<1|1]=rgt[o<<1|1]=sum[o<<1|1]=0;
empty[o<<1]=empty[o<<1|1]=set[o]=0;
set[o<<1]=set[o<<1|1]=1;
}
}
void add(int o,int l,int r){
if(_l<=l&&_r>=r){
set[o]=1;empty[o]=0;
lft[o]=rgt[o]=sum[o]=0;
return;
}else{
int mid=(l+r)>>1;
pushdown(o,l,r);
if(_l<=mid) add(o<<1,l,mid);
if(_r>mid) add(o<<1|1,mid+1,r);
if(sum[o<<1]==mid-l+1){
lft[o]=sum[o<<1]+lft[o<<1|1];
}else lft[o]=lft[o<<1];
if(sum[o<<1|1]==r-mid){
rgt[o]=sum[o<<1|1]+rgt[o<<1];
}else rgt[o]=rgt[o<<1|1];
sum[o]=max(max(sum[o<<1],sum[o<<1|1]),lft[o<<1|1]+rgt[o<<1]);
}
}
void change(int o,int l,int r){
if(_l<=l&&_r>=r){
empty[o]=1;set[o]=0;
lft[o]=rgt[o]=sum[o]=r-l+1;
return;
}else{
int mid=(l+r)>>1;
pushdown(o,l,r);
if(_l<=mid) change(o<<1,l,mid);
if(_r>mid) change(o<<1|1,mid+1,r);
if(sum[o<<1]==mid-l+1){
lft[o]=sum[o<<1]+lft[o<<1|1];
}else lft[o]=lft[o<<1];
if(sum[o<<1|1]==r-mid){
rgt[o]=sum[o<<1|1]+rgt[o<<1];
}else rgt[o]=rgt[o<<1|1];
sum[o]=max(max(sum[o<<1],sum[o<<1|1]),lft[o<<1|1]+rgt[o<<1]);
}
}
int ask(int x){
if(sum[1]<x)return 0;
int o=1,l=1,r=n;
int h=x;
while(l!=r){
pushdown(o,l,r);
if(sum[o<<1]>=x){
o=o<<1;
r=(l+r)>>1;
continue;
}
if(lft[o<<1|1]+rgt[o<<1]>=x) break;
if(sum[o<<1|1]>=x){
o=o<<1|1;
l=((l+r)>>1)+1;
}
}
if(l==r){
return l;
} else{
pushdown(o,l,r);
int mid=(l+r)>>1;
x-=rgt[o<<1];
int k=mid+x;
return k-h+1;
}
}
int main(){
#ifdef LOCAL
freopen("in.txt","r",stdin);
freopen("out.txt","w",stdout);
#endif scanf("%d%d",&n,&m);
mk(1,1,n);
while(m--){
scanf("%d%d",&opt,&x);
if(opt&1){
int k=ask(x);
printf("%d\n",k);
if(k==0) continue;
_l=k;_r=k+x-1;
add(1,1,n);
}else{
_l=x;
scanf("%d",&_r);
_r+=_l-1;
change(1,1,n);
}
}
return 0;
}