POJ 3667 Hotel

题目大意：

1 a:询问是不是有连续长度为a的空房间,有的话住进最左边

2 a b:将[a,a+b-1]的房间清空

思路：线段树的区间合并。

用cov记录区段的状态，-1代表没有被更新，0代表空闲，1代表是有人入住的。

用lmax代表从左端点开始最长的空闲区间，rmax代表从右开始最长的区间。tree代表自己这个区间内拥有的最大区间。

接下来就是两个重要的函数：pushdown  pushup了

void pushup(int num,int mid)
{
    lmax[num]=lmax[num<<1];
    rmax[num]=rmax[num<<1|1];
    if(lmax[num]==(mid-(mid>>1)))lmax[num]+=lmax[num<<1|1];
    if(rmax[num]==(mid>>1))rmax[num]+=rmax[num<<1];
    tree[num]=max(tree[num<<1],max(tree[num<<1|1],rmax[num<<1]+lmax[num<<1|1]));
}

首先在不考虑左右最大区间重叠的情况下，左边最大区间是左儿子的最大左区间，右边的最大区间是又儿子的最大右区间。

当lmax[num]==(mid-(mid>>1))时，就代表左边全部是空闲的，就要考虑一个最大的区间被分成了两边了，就再加上右儿子的最大左区间。

同理。

最后更新最大区间的时候  就要三者做比较    最大的左  最大的右   左右之间的部分。

 

#include <iostream>

#include <cstdio>

#include <algorithm>

#define MAXN 50005

using namespace std;



int tree[MAXN<<2];

int lmax[MAXN<<2];

int rmax[MAXN<<2];

int cov[MAXN<<2];



void pushup(int num,int mid)

{

    lmax[num]=lmax[num<<1];

    rmax[num]=rmax[num<<1|1];

    if(lmax[num]==(mid-(mid>>1)))lmax[num]+=lmax[num<<1|1];

    if(rmax[num]==(mid>>1))rmax[num]+=rmax[num<<1];

    tree[num]=max(tree[num<<1],max(tree[num<<1|1],rmax[num<<1]+lmax[num<<1|1]));

}



void pushdown(int num,int mid)

{

    if(cov[num]!=-1)

    {

        cov[num<<1] = cov[num<<1|1] = cov[num];

        tree[num<<1] = lmax[num<<1] = rmax[num<<1] = cov[num] ? 0: (mid - (mid>>1));

        tree[num<<1|1] = lmax[num<<1|1] = rmax[num<<1|1] = cov[num] ? 0:(mid>>1);

        cov[num]=-1;

    }

}





void build(int num,int l,int r)

{

    lmax[num] = rmax[num] = tree[num] = r - l + 1;

    cov[num] = -1;

    if(r==l)return;

    int mid = (r+l)>>1;

    build(num<<1,l,mid);

    build(num<<1|1,mid+1,r);

}





int query(int num,int s,int e,int w)

{

    if(e==s)

    return s;

    pushdown(num,e-s+1);

    int mid = (s+e)>>1;



    if(tree[num<<1]>=w)return query(num<<1,s,mid,w);

    else if(rmax[num<<1] + lmax[num<<1|1] >=w)return mid + 1 - rmax[num<<1];



    return query(num<<1|1,mid+1,e,w);

}



void update(int num,int s,int e,int l,int r,int v)

{

    if(l<=s && e<=r)

    {

        tree[num] = lmax[num] = rmax[num] = v ? 0 : e-s+1;

        cov[num] = v;

        return ;

    }

    pushdown(num,e-s+1);

    int mid = (e+s)>>1;

    if(l<=mid)update(num<<1,s,mid,l,r,v);

    if(r>mid)update(num<<1|1,mid+1,e,l,r,v);

    pushup(num,e-s+1);

}





int main()

{

    int n,m;



        scanf("%d%d",&n,&m);

        build(1,1,n);

        for(int i=1;i<=m;i++)

        {

            int a,b,c;

            scanf("%d",&a);



            if(a==1)

            {

                scanf("%d",&b);

                if(tree[1]<b)puts("0");

                else {

                int p=query(1,1,n,b);

                printf("%d\n",p);

                update(1,1,n,p,p+b-1,1);

                }

            }

            else

            {

                scanf("%d%d",&b,&c);

                update(1,1,n,b,b+c-1,0);

            }

        }

        return 0;

}