【线段树区间合并】POJ3667-Hotel

【题意】

一段区间初始均为可行。有两个操作：

1→找出长度为w的一段可行区间，如果存在则返回这个可行区间最靠左的情况，并将该区间设为不可行；

2→将区间[a,b]设为可行区间。

【思路】

经典的线段树合并，代码依旧用的是神犇的线段树模板。详见注释。

【错误点】

延迟标记的时候，忘记把cover清为-1了，导致RE！

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<cstdlib>
 using namespace std;
 #define lson l,m,rt<<1
 #define rson m+1,r,rt<<1|1
 const int MAXN=+;
 int cover[MAXN<<];//-1表示当前没有覆盖标记,1表示均覆盖为不可行，0表示均覆盖为可行
 int lsum[MAXN<<];//该区间从左起连续的可用区间长度的最大值
 int msum[MAXN<<];//该区间中连续的可用区间长度的最大值
 int rsum[MAXN<<];//该区间从右起连续的可用区间长度的最大值 

 void PushUp(int rt,int m)
 {
     lsum[rt]=lsum[rt<<];
     if (lsum[rt]==m-(m>>)) lsum[rt]+=lsum[rt<<|];
     /*如果左孩子全部为可用区间，那么加上右孩子的左端*/
     rsum[rt]=rsum[rt<<|];
     if (rsum[rt]==m>>) rsum[rt]+=rsum[rt<<];
     /*同上*/
     msum[rt]=max(max(msum[rt<<],msum[rt<<|]) , rsum[rt<<]+lsum[rt<<|]);
     /*该区间的可用区间可能是：左孩子最大的可用区间、有孩子最大的可用区间，和跨越左右孩子加在一起的可用区间*/
 }

 void PushDown(int rt,int m)
 {
     if (cover[rt]!=-)
     {
         cover[rt<<]=cover[rt<<|]=cover[rt];
         if (cover[rt]==)
         {
             msum[rt<<]=lsum[rt<<]=rsum[rt<<]=;
             msum[rt<<|]=lsum[rt<<|]=rsum[rt<<|]=;
         }
         else
         {
             msum[rt<<]=lsum[rt<<]=rsum[rt<<]=m-(m>>);
             msum[rt<<|]=lsum[rt<<|]=rsum[rt<<|]=m>>;
         }
         cover[rt]=-;
         /*千万不要忘记将rt清为-1*/
     }
 }

 int query(int w,int l,int r,int rt)
 {
     if (l==r) return l;
     PushDown(rt,r-l+);
     int m=(l+r)>>;
     if (msum[rt<<]>=w) return(query(w,lson));
     /*由于要找最左边的区间，按照左孩子、跨越两者、有孩子的顺序查找*/
     if (rsum[rt<<]+lsum[rt<<|]>=w) return(m-rsum[rt<<]+);
     return(query(w,rson));
 }

 void update(int L,int R,int o,int l,int r,int rt)
 {
     if (L<=l && r<=R)
     {
         cover[rt]=o;
         if (o==) msum[rt]=lsum[rt]=rsum[rt]=;
             else msum[rt]=lsum[rt]=rsum[rt]=r-l+;
         return;
     }
     PushDown(rt,r-l+);//这里是l和r，不要写成L和R
     int m=(l+r)>>;
     if (L<=m) update(L,R,o,lson);
     if (m<R)  update(L,R,o,rson);
     PushUp(rt,r-l+);
 }

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

 int main()
 {
     int n,m;
     scanf("%d%d",&n,&m);
     build(,n,);
     for (int i=;i<m;i++)
     {
         int op;
         scanf("%d",&op);
         if (op==)
         {
             int w;
             scanf("%d",&w);
             if (msum[]<w) cout<<<<endl;
             /*如果根的可用区间已经小于w，那么一定是找不到长度为w的可用区间*/
                 else
                 {
                       int p=query(w,,n,);
                       cout<<p<<endl;
                       update(p,p+w-,,,n,);
                 }
         }
         else
         {
             int u,v;
             scanf("%d%d",&u,&v);
             update(u,u+v-,,,n,);
         }
     }
     return ;
 }