POJ1823hotel线段树

这题话说以我的能力，实在写不出来，后来看了瓜神的博客，才知道这题写法，觉得自己对线段树的区间合并理解还是不够

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <climits>
#include <string>
#include <iostream>
#include <map>
#include <cstdlib>
#include <list>
#include <set>
#include <queue>
#include <stack>
using namespace std;
#define INF 0xfffffff
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define maxn 50010
int col[maxn<<2],sum[maxn<<2],lsum[maxn<<2],rsum[maxn<<2];
void pushdown(int rt,int m)
{
    if(col[rt]!=-1)
    {
        col[rt<<1]=col[rt<<1|1]=col[rt];
        sum[rt<<1]=lsum[rt<<1]=rsum[rt<<1]=col[rt]?0:(m-(m>>1));
        sum[rt<<1|1]=lsum[rt<<1|1]=rsum[rt<<1|1]=col[rt]?0:m>>1;
        col[rt]=-1;
    }
}
void pushup(int rt,int m)
{
    lsum[rt] = lsum[rt << 1];
    rsum[rt] = rsum[rt << 1 | 1];
    if (lsum[rt] == (m - (m >> 1))) lsum[rt] += lsum[rt << 1 | 1];
    if (rsum[rt] == (m >> 1)) rsum[rt] += rsum[rt << 1];
    sum[rt] = max(rsum[rt << 1] + lsum[rt << 1 | 1], max(sum[rt << 1], sum[rt << 1 | 1]));
}
void build(int l,int r,int rt)
{
    col[rt]=-1;
    sum[rt]=lsum[rt]=rsum[rt]=r-l+1;
    if(l==r)
    {
        return ;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
}
void update(int L,int R,int add,int l,int r,int rt)
{
    if(L<=l&&R>=r)
    {
        sum[rt]=lsum[rt]=rsum[rt]=add?0:r-l+1;
        col[rt]=add;
        return ;
    }
    pushdown(rt,r-l+1);
    int m=(l+r)>>1;
    if(L<=m) update(L,R,add,lson);
    if(R>m) update(L,R,add,rson);
    pushup(rt,r-l+1);
}
int query(int key,int l,int r,int rt)
{
    if(l==r) return l;
    int m=(l+r)>>1;
    pushdown(rt,r-l+1);
    if(sum[rt<<1]>=key) return query(key,lson);
    else if(rsum[rt<<1]+lsum[rt<<1|1]>=key) return m-rsum[rt<<1]+1;
    else return query(key,rson);
}
int main()
{
    int n,m,t;
    int a,b;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        build(1,n,1);
        while(m--)
        {
            scanf("%d",&t);
            if(t==1)
            {
                scanf("%d",&a);
                if(a>sum[1])
                {
                    printf("0\n");
                    continue;
                }
                int pos=query(a,1,n,1);
                printf("%d\n",pos);
                update(pos,pos+a-1,1,1,n,1);
            }
            else
            {
                scanf("%d%d",&a,&b);
                update(a,a+b-1,0,1,n,1);
            }
        }
    }
    return 0;
}