POJ 3667 Hotel (线段树区间合并)

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

最初给你n间空房，m个操作：

　　操作1 a 表示检查是否有连续的a间空房，输出最左边的空房编号，并入住a间房间。

　　操作2 a b 表示将编号为a之后的b间房间清空。

典型的区间合并问题，这位大牛讲的更清楚：http://www.cnblogs.com/yewei/archive/2012/05/05/2484471.html

代码如下：

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int MAXN = 5e4 + 5;
//sum表示这个区间的最大数，lsum表示从这个区间最左边开始连续的最大数，rsum与lsum方向相反
struct data {
    int l , r , lsum , rsum , sum , cover;
}T[MAXN << 2];

void build(int p , int l , int r) {
    int mid = (l + r) >> 1;
    T[p].lsum = T[p].rsum = T[p].sum = r - l + 1;
    T[p].cover = -1;
    T[p].l = l , T[p].r = r;
    if(l == r)
        return ;
    build(p << 1 , l , mid);
    build((p << 1)|1 , mid + 1 , r);
}

void Pushdown(int p) {
    if(T[p].cover != -1) {
        int ls = p << 1 , rs = (p << 1)|1;
        T[ls].cover = T[rs].cover = T[p].cover;
        T[ls].lsum = T[ls].rsum = T[ls].sum = T[p].cover ? 0 : T[ls].r - T[ls].l + 1;
        T[rs].lsum = T[rs].rsum = T[rs].sum = T[p].cover ? 0 : T[rs].r - T[rs].l + 1;
        T[p].cover = -1;
    }
}

void Pushup(int p) {
    T[p].lsum = T[p << 1].lsum;
    T[p].rsum = T[(p << 1)|1].rsum;
    if(T[p].lsum == T[p << 1].r - T[p << 1].l + 1)
        T[p].lsum += T[(p << 1)|1].lsum;
    if(T[p].rsum == T[(p << 1)|1].r - T[(p << 1)|1].l + 1)
        T[p].rsum += T[p << 1].rsum;
    int temp = max(T[(p << 1)|1].sum , T[p << 1].rsum + T[(p << 1)|1].lsum);
    T[p].sum = max(T[p << 1].sum , temp);
}

int query(int p , int len) {
    int mid = (T[p].l + T[p].r) >> 1;
    if(T[p].l == T[p].r) {
        return T[p].l;
    }
    Pushdown(p);
    if(T[p << 1].sum >= len) { //左孩子区间的最大数，优先选左孩子
        return query(p << 1 , len);
    }
    else if(T[p << 1].rsum + T[(p << 1)|1].lsum >= len) { //其次要是左孩子的rsum和右孩子的lsum 加起来大于len
        return mid - T[p << 1].rsum + 1;
    }
    else {
        return query((p << 1)|1 , len); //最后就是右孩子
    }
}

void updata(int p , int l , int r , int flag) {
    int mid = (T[p].l + T[p].r) >> 1;
    if(T[p].l == l && T[p].r == r) {
        T[p].sum = T[p].lsum = T[p].rsum = flag ? 0 : r - l + 1;
        T[p].cover = flag;
        return ;
    }
    Pushdown(p);
    if(r <= mid) {
        updata(p << 1 , l , r , flag);
    }
    else if(l > mid) {
        updata((p << 1)|1 , l , r , flag);
    }
    else {
        updata(p << 1 , l , mid , flag);
        updata((p << 1)|1 , mid + 1 , r , flag);
    }
    Pushup(p);
}

int main()
{
    int n , m , choose , u , v;
    while(~scanf("%d %d" , &n , &m)) {
        build(1 , 1 , n);
        while(m--) {
            scanf("%d" , &choose);
            if(choose == 1) {
                scanf("%d" , &u);
                if(T[1].sum < u)
                    printf("0
");
                else {
                    int pos = query(1 , u);
                    printf("%d
" , pos);
                    updata(1 , pos , pos + u - 1 , 1);
                }
            }
            else {
                scanf("%d %d" , &u , &v);
                updata(1 , u , min(u + v - 1 , n) , 0);
            }
        }
    }
    return 0;
}