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