POJ 3667 hotel 线段树区间合并

有n房间，序号1-n，m次询问，询问“1 b”代表入住b人，你需要找一段连续的6间房间，若有返回序号最小的房间序号，若没有，返回0；询问“2 b c”代表清理房间b-c,清理之后b-c房间可以入住。

思路：记录某区间的最大连续房间个数、区间从左边第一个开始的连续空房间个数，右边最后一个往前连续的空房间个数分别用sum、lsum、rsum表示。用cover表示某区间是否被入住，0表示全部未入住，1表示全部入住，-1表示部分入住。

见代码

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
using namespace std;
#define maxn 121111
int rsum[maxn<<2],lsum[maxn<<2],sum[maxn<<2],cover[maxn<<2];
void build(int l,int r,int node)
{
    sum[node]=lsum[node]=rsum[node]=r-l+1;//一开始，所以房间均为空房，
    cover[node]=0;
    if(r==l)
        return ;
    int mid=(l+r)>>1;
    build(l,mid,node<<1);
    build(mid+1,r,node<<1|1);

}
void pushdown(int node,int len)
{
    if(cover[node]!=-1)
    {
        cover[node<<1]=cover[node<<1|1]=cover[node];
        sum[node<<1]=lsum[node<<1]=rsum[node<<1]=cover[node]?0:len-(len>>1);
        sum[node<<1|1]=lsum[node<<1|1]=rsum[node<<1|1]=cover[node]?0:(len>>1);
        cover[node]=-1;
    }
}
void pushup(int node,int len)
{
    lsum[node]=lsum[node<<1];//当前区间左1起的连续个数一定大于等于左子区间的左1起的连续区间
    rsum[node]=rsum[node<<1|1];//同理
    if(lsum[node]==len-(len>>1))//左子区间全为空，此时合并右子区间的左1起的空房子
        lsum[node]+=lsum[node<<1|1];
    if(rsum[node]==(len>>1))//同理
        rsum[node]+=rsum[node<<1];
    sum[node]=max(lsum[node<<1|1]+rsum[node<<1],max(sum[node<<1],sum[node<<1|1]));//当前区间的最大连续房子的个数等于左1起的连续空房子个数、右未往前的连续空房子个数、中间的跨区间空房子个数中的最大值
}
int cal(int a,int l,int r,int node)
{
    if(l==r)
        return l;
    pushdown(node,r-l+1);//需要对当前区间进行操作，先对当前区间拆分
    int mid=(l+r)>>1;
    if(sum[node<<1]>=a)//左子区间的连续空房子大于等于所需的连续空房子
        return cal(a,l,mid,node<<1);
    else if((rsum[node<<1]+lsum[node<<1|1])>=a)//跨区间左子区间右边+右子区间左边的房子满足所需连续的房子
        return mid-rsum[node<<1]+1;
    return cal(a,mid+1,r,node<<1|1);//右子区间的连续空房子大于等于所需的连续空房子

}
void update(int L,int R,int c,int l,int r,int node)//c=1，入住后更新，c=0，清理房间更新
{
    if(L<=l&&R>=r)
    {
        cover[node]=c;
        rsum[node]=lsum[node]=sum[node]=c?0:r-l+1;
        //printf("A%d %d %d %d %dA\n",node,l,r,c,lsum[node]);
        return ;
    }
    pushdown(node,r-l+1);//对当前区间进行操作先拆分
    int mid=(l+r)>>1;
    if(L<=mid)
        update(L,R,c,l,mid,node<<1);
    if(R>mid)
        update(L,R,c,mid+1,r,node<<1|1);
    pushup(node,r-l+1);//更新祖宗及合并
}
int main()
{
    int  n,m;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        build(1,n,1);
        int a,b,c;
        while(m--)
        {
            scanf("%d",&a);
            if(a==1)
            {
                scanf("%d",&b);
                if(sum[1]<b)
                    puts("0");
                else
                {
                    int p=cal(b,1,n,1);
                    //for(int i=1;i<=n*4;i++)
                       // printf("%d %d %d %d %d\n",i,sum[i],lsum[i],rsum[i],cover[i]);
                    printf("%d\n",p);
                    update(p,p+b-1,1,1,n,1);//入住后更新
                }
            }
            else
            {
                scanf("%d%d",&b,&c);
                update(b,b+c-1,0,1,n,1);
            }
        }
    }
    return 0;
}