POJ-3667：Hotel（线段树区间合并）

题目链接：点击打开链接

题目大意：

原题为POJ 3667 Hotel问题，题目大意是有一个无限长的长廊，然后有1到n个房间，每次的操作有两种。

第一种操作，是输入人的个数，尽量向左给其连续的房间并输出第一个房间号。

第二个操作是给你一段区间，代表此区间的人移出房间。

解题思路：

这是我做的第一道线段树区间更新的题，也花了较长的时间来理解。其中区间更新要用到延迟更新，不然会tle，还有区间合并，就是左最长右最长区间最长一层层向上维护即可，以下贴代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
int num[500000];
int a[5005];
int INF=1e9+5;
int n,sum,m,ans;
struct node
{
    int l,r,mid;
    int llen,rlen,tlen;
    int flag;
}t[500000];
int mark[500000];    //延时更新标记数组
void pushdown(int rt,int len)  //延时更新操作
{
    if(mark[rt]!=-1)
    {
        mark[rt<<1]=mark[rt<<1|1]=mark[rt];
        if(mark[rt]==1)
        {
            t[rt<<1].rlen=t[rt<<1].llen=t[rt<<1].tlen=len-len/2;
            t[rt<<1|1].rlen=t[rt<<1|1].llen=t[rt<<1|1].tlen=len/2;
        }
        else
        {
            t[rt<<1].rlen=t[rt<<1].llen=t[rt<<1].tlen=0;
            t[rt<<1|1].rlen=t[rt<<1|1].llen=t[rt<<1|1].tlen=0;
        }
        mark[rt]=-1;
    }
}
void pushup(int rt)         //区间更新
{
    if(t[rt<<1].llen!=t[rt<<1].r-t[rt<<1].l+1)
        t[rt].llen=t[rt<<1].llen;
    else
        t[rt].llen=t[rt<<1].llen+t[rt<<1|1].llen;
    if(t[rt<<1|1].rlen!=t[rt<<1|1].r-t[rt<<1|1].l+1)
        t[rt].rlen=t[rt<<1|1].rlen;
    else
        t[rt].rlen=t[rt<<1|1].rlen+t[rt<<1].rlen;
    t[rt].tlen=max(max(t[rt<<1].tlen,t[rt<<1].rlen+t[rt<<1|1].llen),t[rt<<1|1].tlen);
}
void build(int l,int r,int rt)          //建立线段树
{
    t[rt].l=l;
    t[rt].r=r;
    t[rt].mid=(l+r)>>1;
    t[rt].llen=t[rt].rlen=t[rt].tlen=r-l+1;
    if(l==r&&l!=0)
    {
        t[rt].llen=t[rt].rlen=t[rt].tlen=1;
        t[rt].flag=0;
        return ;
    }
    build(l,t[rt].mid,rt<<1);
    build(t[rt].mid+1,r,rt<<1|1);
    pushup(rt);
}
void update(int l,int r,int rt,int flag)        //更新l到r的区间的值
{
    if(l<=t[rt].l&&r>=t[rt].r)
    {
        mark[rt]=flag;
        if(flag)
            t[rt].rlen=t[rt].llen=t[rt].tlen=t[rt].r-t[rt].l+1;
        else
            t[rt].rlen=t[rt].llen=t[rt].tlen=0;
        return ;
    }
    pushdown(rt,t[rt].r-t[rt].l+1);
    if(l<=t[rt].mid) update(l,r,rt<<1,flag);
    if(r>t[rt].mid) update(l,r,rt<<1|1,flag);
    pushup(rt);
}
int sear(int n,int rt)      //寻找能容纳n个人的最大左区间
{
    if(t[rt].l==t[rt].r)
        return 1;
    pushdown(rt,t[rt].r-t[rt].l+1);
    if(t[rt<<1].tlen>=n)
        return  sear(n,rt<<1);
    else if(t[rt<<1].rlen+t[rt<<1|1].llen>=n)
        return  t[rt].mid-t[rt<<1].rlen+1;
    else
        return sear(n,rt<<1|1);
}
void print(int rt)
{
    if(t[rt].l==t[rt].r)
    {
        printf("%d\n",t[rt].tlen);
        return ;
    }
    print(rt<<1);
    print(rt<<1|1);
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        memset(mark,-1,sizeof(mark));
        memset(t,0,sizeof(t));
        build(1,n,1);
        for(int i=1;i<=m;i++)
        {
            int qu;
            scanf("%d",&qu);
            if(qu==1)
            {
                int p;
                scanf("%d",&p);
                if(p>t[1].tlen)
                    printf("0\n");
                else
                {
                    ans=sear(p,1);
                    update(ans,ans+p-1,1,0);
                    printf("%d\n",ans);
                }
            }
            if(qu==2)
            {
                int p,q;
                scanf("%d%d",&p,&q);
                update(p,p+q-1,1,1);
            }
        }
    }
}