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 + ;
 //sum表示这个区间的最大数，lsum表示从这个区间最左边开始连续的最大数，rsum与lsum方向相反
 struct data {
     int l , r , lsum , rsum , sum , cover;
 }T[MAXN << ];

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

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

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

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

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

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