poj3667

线段树好难啊

题意是 给1到n个区间m次操作 如果是1的换就是入住k个人 要求连续房间并且房间号最小 能入住输出第一个房间号如果不能入住则输出-1  如果是2就是一个区间内的客人退房

区间合并的问题  lm表示从当前区间左边开始连续房间的个数 rm右边 cm中间

主要就是PU和PD操作  query和update都比较好理解

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define MAXN 100000
#define ls rt<<1
#define rs rt<<1|1
struct Node
{
    int l,r,lm,rm,col,cm;
    int len()
    {
        return r-l+1;
    }
    void doit()
    {
        lm=rm=cm=col==1?0:r-l+1;
        return;
    }
}N[MAXN<<2];
void build (int rt,int l,int r)
{
    N[rt].l=l;N[rt].r=r;
    N[rt].lm=N[rt].rm=N[rt].cm=r-l+1;
    N[rt].col=-1;
    if(l==r)
    return;
    int mid=(l+r)/2;
    build(ls,l,mid);
    build(rs,mid+1,r);
}
void PU(int rt)
{
    N[rt].lm=N[ls].lm==N[ls].len()?N[ls].len()+N[rs].lm:N[ls].lm;
    N[rt].rm=N[rs].rm==N[rs].len()?N[rs].len()+N[ls].rm:N[rs].rm;
    N[rt].cm=max(N[ls].rm+N[rs].lm,max(N[ls].cm,N[rs].cm));
}
void PD(int rt)
{
    if(N[rt].col==-1||N[rt].len()==1)
    return;
    N[ls].col=N[rs].col=N[rt].col;
    N[rt].col=-1;
    N[ls].doit();
    N[rs].doit();
}
void  update(int rt,int l,int r,int val)
{
    if(l<=N[rt].l&&r>=N[rt].r)
    {
        N[rt].col=val;
        N[rt].doit();
        return;
    }
    PD(rt);
    int mid=(N[rt].l+N[rt].r)/2;
    if(mid>=l)
    update(ls,l,r,val);
    if(mid<r)
    update(rs,l,r,val);
    PU(rt);
}
int query(int rt,int k)
{
    if(N[rt].len()==1)
    return k==1;
    PD(rt);
    if(N[ls].cm>=k)
    return query(ls,k);
    if(N[ls].rm+N[rs].lm>=k)
    return N[ls].r-N[ls].rm+1;
    if(N[rs].cm>=k)
    return query(rs,k);
    return 0;
}
void find(int rt)
{
    if(N[rt].len()==1)
    {
        printf("%d ",N[rt].col);
        return;
    }
    PD(rt);
    find(ls);
    find(rs);
}
int main()
{
    int n,m;
    scanf ("%d%d",&n,&m);
    {
        build(1,1,n);
        int op;int l,r;
        for(int i=0;i<m;i++)
        {
            scanf ("%d",&op);
            if(op==1)
            {
                scanf("%d",&l);
                int s=query(1,l);
                if(s<=0)
                {
                    printf("0\n");
                }
                else{
                    printf("%d\n",s);
                    update(1,s,s+l-1,1);
                    //find(1);
                }
            }
            else
            {
                scanf ("%d%d",&l,&r);
                update(1,l,l+r-1,0);
                //find(1);
            }
        }
    }
}