Poj 3667 Hotel 成段更新，区间合并，区间求和

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

题目大意：一个旅馆，有N个房间，初始状态全为空，现在有客人要入住，但是一批客人只能住连在一起的几个房间，现在给你多批客人，问每批客人能入住的第一个房间的最小编号，如果不能入住输出0

解题思路：首先线段树应包含4个域，分别是这个区间最长连续的房间数msum，这个区间从左边起的空房间数lsum，这个区间从右边起的空房间数rsum，和标记该房间是否有人mark，然后更新的时候注意一个区间的rsum和相邻区间的lsum是想连的，所以向上更新时可能最长空房间数是由它们两个相加得到的，即区间合并。

代码如下：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 50005
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
int msum[N<<2],lsum[N<<2],rsum[N<<2],mark[N<<2];
void pushup(int rt,int m)
{
    lsum[rt]=lsum[rt<<1]==m-(m>>1)?(lsum[rt<<1]+lsum[rt<<1|1]):(lsum[rt<<1]);
    rsum[rt]=rsum[rt<<1|1]==(m>>1)?(rsum[rt<<1|1]+rsum[rt<<1]):(rsum[rt<<1|1]);
    msum[rt]=max(rsum[rt<<1]+lsum[rt<<1|1],max(msum[rt<<1],msum[rt<<1|1]));
}
void pushdown(int rt,int m)
{
    if(mark[rt]>=0)
    {
        mark[rt<<1]=mark[rt<<1|1]=mark[rt];
        lsum[rt<<1]=rsum[rt<<1]=msum[rt<<1]=mark[rt]?0:(m-(m>>1));
        lsum[rt<<1|1]=rsum[rt<<1|1]=msum[rt<<1|1]=mark[rt]?0:(m>>1);
        mark[rt]=-1;
    }
}
void build(int l,int r,int rt)
{
    lsum[rt]=rsum[rt]=msum[rt]=r-l+1;
    mark[rt]=-1;
    if(l==r) return;
    int m=(l+r)>>1;
    build(lson);
    build(rson);
}
void update(int L,int R,int c,int l,int r,int rt)
{
    if(L<=l&&R>=r)
    {
        msum[rt]=lsum[rt]=rsum[rt]=c?0:r-l+1;
        mark[rt]=c;
        return;
    }
    pushdown(rt,r-l+1);
    int m=(l+r)>>1;
    if(L<=m) update(L,R,c,lson);
    if(R>m) update(L,R,c,rson);
    pushup(rt,r-l+1);
}
int query(int x,int l,int r,int rt)
{
    if(l==r) return l;
    pushdown(rt,r-l+1);
    int m=(l+r)>>1;
    if(x<=msum[rt<<1]) return query(x,lson);
    else if(x<=rsum[rt<<1]+lsum[rt<<1|1]) return m-rsum[rt<<1]+1;
    else return query(x,rson);
    pushup(rt,r-l+1);
}
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        build(1,n,1);
        for(int i=0;i<m;i++)
        {
            int op,a,b;
            scanf("%d",&op);
            if(op==1)
            {
                scanf("%d",&a);
                if(msum[1]<a) printf("0\n");
                else
                {
                    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;
}