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);            }        }    }}