POJ 3667 Hotel 线段树区间合并

题意

一开始有1~n个空房间

对应2个操作 

i）问有没有连续的长度为x的x间房如果有占满最左边的那个区间，输出这个区间的起点

ii）把区间x~x+y-1的房间清空

一个很明显的区间合并具体注意点写在注释里了

ACcode：

#include <cstdio>
#include <algorithm>
#define tmp (st<<1)
#define mid ((l+r)>>1)
#define lson l,mid,tmp
#define rson mid+1,r,tmp|1
#define len (r-l+1)
#define maxn 50002
using namespace std;
int sum[maxn<<2],lsum[maxn<<2],rsum[maxn<<2],col[maxn<<2];
inline void push_up(int st,int m){
    lsum[st]=lsum[tmp];
    rsum[st]=rsum[tmp|1];
    if(lsum[st]==m-(m>>1))lsum[st]+=lsum[tmp|1];
    if(rsum[st]==m>>1)rsum[st]+=rsum[tmp];
    sum[st]=max(max(sum[tmp],sum[tmp|1]),lsum[tmp|1]+rsum[tmp]);
}
inline void push_down(int st,int m){
    if(col[st]!=-1){
        col[tmp]=col[tmp|1]=col[st];
        sum[tmp]=lsum[tmp]=rsum[tmp]=col[st]*(m-(m>>1));
        sum[tmp|1]=lsum[tmp|1]=rsum[tmp|1]=col[st]*(m>>1);
        col[st]=-1;
    }
}
inline void build(int l,int r,int st){
    col[st]=-1;
    sum[st]=lsum[st]=rsum[st]=len;
    if(l==r)return;
    build(lson);
    build(rson);
}
inline void updata(int L,int R,int add,int l,int r,int st){
    if(L<=l&&r<=R){
        col[st]=add;
        sum[st]=lsum[st]=rsum[st]=add*len;
        return ;
    }
    push_down(st,len);
    if(L<=mid)updata(L,R,add,lson);
    if(mid<R)updata(L,R,add,rson);
    push_up(st,len);
}
inline int query(int x,int l,int r,int st){
    if(l==r)return l;
    push_down(st,len);
    if(sum[tmp]>=x)return query(x,lson);///左子树满足
    else if(rsum[tmp]+lsum[tmp|1]>=x)return mid-rsum[tmp]+1;///左子树的右边区间和右子树的左区间满足条件
    return query(x,rson);
}
int main(){
    int n,q;
    while(~scanf("%d%d",&n,&q)){
        build(1,n,1);
        int z,x,y;
        while(q--){
            scanf("%d%d",&z,&x);
            if(z==1){
                if(sum[1]<x){printf("0\n");continue;}///总区间就不满足
                int temp=query(x,1,n,1);
                printf("%d\n",temp);
                updata(temp,temp+x-1,0,1,n,1);
            }
            else{
                scanf("%d",&y);
                y=x+y-1;
                updata(x,y,1,1,n,1);
            }
        }
    }
    return 0;
}
/*
10 6
1 3
1 3
1 3
1 3
2 5 5
1 6
*/