poj 3667 Hotel（线段树，区间合并）

HH神总结的线段树专辑：https://wenku.baidu.com/view/71fc1659ba1aa8114431d97b.html
里面的例题，看懂代码后。基本也就把代码背下来了。。。
自己再写一遍，基本就是在默写。。

#include <cstdio>
#include <cstring>
#include <algorithm>

#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1

const int MAXN = 50010;
//区间为l,r  lsum表示从l向右起有多少连续的空房间，rsum表示从r向左有多少连续的空房间
//msum表示l,r整个区间最长的连续空房间
int lsum[MAXN<<2],rsum[MAXN<<2],msum[MAXN<<2];
int cover[MAXN<<2];

void build(int l,int r, int rt)
{
    lsum[rt] = rsum[rt] = msum[rt] = r-l+1;
    cover[rt] = -1;
    if(l == r) return;
    int m = (l+r)>>1;
    build(lson);
    build(rson);
}

void pushDown(int rt, int len)
{
    if(cover[rt] != -1)
    {
        cover[rt<<1] = cover[rt<<1|1] = cover[rt];
        lsum[rt<<1] = rsum[rt<<1] = msum[rt<<1] = cover[rt] ? 0: len-(len>>1);
        lsum[rt<<1|1] = rsum[rt<<1|1] = msum[rt<<1|1] = cover[rt] ? 0 : len>>1;
        cover[rt] = -1;
    }
}

void pushUp(int rt, int len)
{
    //根据子节点更新父节点
    lsum[rt] = lsum[rt<<1];
    rsum[rt] = rsum[rt<<1|1];
    //合并相邻区间
    if(lsum[rt<<1] == len-(len>>1)) lsum[rt] += lsum[rt<<1|1];
    if(rsum[rt<<1|1] == (len>>1)) rsum[rt] += rsum[rt<<1];

    msum[rt] = std::max(rsum[rt<<1]+lsum[rt<<1|1],std::max(msum[rt<<1],msum[rt<<1|1]));
}

int query(int l, int r, int rt, int len)
{
    if(l == r) return l;
    pushDown(rt,r-l+1);
    int m = (l+r)>>1;
    //如果左孩子节点满足要求，就搜索做孩子
    if(msum[rt<<1] >= len) return query(lson,len);
    //否则的话，看左孩子的右区间和右孩子的左区间合并后是否符合条件
    //符合就返回
    else if(rsum[rt<<1]+lsum[rt<<1|1] >= len)
        return m-rsum[rt<<1]+1;
    //最后再搜索右孩子
    return query(rson,len);
}

void update(int L, int R, int c, int l, int r , int rt)
{
    if(l >= L && r <= R)
    {
        lsum[rt] = msum[rt] = rsum[rt] = c ? 0:r-l+1;
        cover[rt] = c;
        return;
    }
    pushDown(rt,r-l+1);
    int m =(l+r)>>1;
    if(m  >= L) update(L,R,c,lson);
    if(R > m) update(L,R,c,rson);
    pushUp(rt,r-l+1);
}

int main()
{
    int n,m,op,a,b;
    scanf("%d %d",&n,&m);
    build(1,n,1);
    while(m--)
    {
        scanf("%d",&op);
        if(op == 1)
        {
            scanf("%d",&a);
            if(a > msum[1])
                printf("0\n");
            else
            {
                int p = query(1,n,1,a);
                printf("%d\n",p);
                update(p,p+a-1,1,1,n,1);
            }
        }
        else
        {
            scanf("%d %d",&a,&b);
            update(a,a+b-1,0,1,n,1);
        }
    }
    return 0;
}