首页 > 代码库 > BZOJ 1593: [Usaco2008 Feb]Hotel 旅馆
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
题解：
在线段树上二分第一个>=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;}
BZOJ 1593: [Usaco2008 Feb]Hotel 旅馆