POJ 3667 Hotel（线段树 区间合并 模板题）

题目：http://poj.org/problem?id=3667

大意：

hotel有n个房间，每次有两种操作：

1.查询并更新：从最左边往里连续住d个人（保证住的下），输出最左边的房号，住不下输出0   

2.更新：从第x个位置开始搬出x+d-1个人

#include<cstdio>
#include<algorithm>
#define lson rt<<1,l,mid
#define rson rt<<1|1,mid+1,r
using namespace std;
const int maxn=1e5+10;

int n,m;
struct node{
    int l,r;
    int lsum;  //从该区间最左端开始（到右端点）共有lsum个连续空房间
    int rsum;  //从该区间最右端开始（到左端点）共有rsum个连续空房间
    int sum;   //该区间最多有sum个连续空房间
}t[maxn<<2];
int lazy[maxn<<2];

void down(int rt){
    if(lazy[rt]!=-1){
        lazy[rt<<1]=lazy[rt<<1|1]=lazy[rt];
        if(lazy[rt]){  //1住满，该区间无空房间
            t[rt<<1].lsum=t[rt<<1].rsum=t[rt<<1].sum=0;
            t[rt<<1|1].lsum=t[rt<<1|1].rsum=t[rt<<1|1].sum=0;
        }
        else{          //0全搬走，空房间数传给子树
            t[rt<<1].lsum=t[rt<<1].rsum=t[rt<<1].sum=t[rt<<1].r-t[rt<<1].l+1;
            t[rt<<1|1].lsum=t[rt<<1|1].rsum=t[rt<<1|1].sum=t[rt<<1|1].r-t[rt<<1|1].l+1;
        }
        lazy[rt]=-1;
    }
}
void up(int rt,int l,int r){
    t[rt].lsum=t[rt<<1].lsum;
    t[rt].rsum=t[rt<<1|1].rsum;
    int mid=(l+r)>>1;                       //若左子树全空，加上右子树的空房间
    if(t[rt].lsum==mid-l+1) t[rt].lsum+=t[rt<<1|1].lsum;
    if(t[rt].rsum==r-mid ) t[rt].rsum+=t[rt<<1].rsum;  //左，右，还是中间部分最大
    t[rt].sum=max(t[rt<<1].rsum+t[rt<<1|1].lsum,max(t[rt<<1].sum,t[rt<<1|1].sum));
}

void build(int rt,int l,int r){
    t[rt].l=l,t[rt].r=r;
    t[rt].sum=t[rt].lsum=t[rt].rsum=r-l+1; //开始都是空房
    lazy[rt]=-1;
    if(l==r){
        return ;
    }
    int mid=(l+r)>>1;
    build(lson),build(rson);
}

void update(int rt,int l,int r,int k){
    if(l<=t[rt].l&&r>=t[rt].r){
        lazy[rt]=k;
        if(k)
            t[rt].sum=t[rt].lsum=t[rt].rsum=0;  //1的话住进来，该区间没有空房间
        else
            t[rt].sum=t[rt].lsum=t[rt].rsum=r-l+1;
        return ;
    }
    down(rt);
    int mid=(t[rt].l+t[rt].r)>>1;
    if(mid>=r) update(rt<<1,l,r,k);
    else if(mid+1<=l) update(rt<<1|1,l,r,k);
    else{
        update(rt<<1,l,mid,k);
        update(rt<<1|1,mid+1,r,k);
    }        //区间合并一定要这么写
    up(rt,t[rt].l,t[rt].r);
}

int query(int rt,int l,int r,int d){  //找最左边位置，分别从左，中间，右找
    if(l==r)
        return l;
    int mid=(l+r)>>1;
    down(rt);
    if(t[rt<<1].sum>=d)
         return query(rt<<1,l,mid,d);
    else if(t[rt<<1].rsum+t[rt<<1|1].lsum>=d)
         return mid-t[rt<<1].rsum+1;
    else
         return query(rt<<1|1,mid+1,r,d);
}
int main(){
    while(scanf("%d%d",&n,&m)==2){
        build(1,1,n);
        while(m--){
            int f,x,d;
            scanf("%d",&f);
            if(f==1){
                scanf("%d",&d);
                if(d>t[1].sum) printf("0\n");
                else{
                    int l=query(1,1,n,d);
                    printf("%d\n",l);
                    update(1,l,l+d-1,1); //查到最左端的位置，在这个区间加人
                }
            }
            else {
                scanf("%d%d",&x,&d);
                update(1,x,x+d-1,0);
            }
        }
    }
    return 0;
}