题目链接:http://poj.org/problem?id=3667
题目大意:一个旅馆,有N个房间,初始状态全为空,现在有客人要入住,但是一批客人只能住连在一起的几个房间,现在给你多批客人,问每批客人能入住的第一个房间的最小编号,如果不能入住输出0
解题思路:首先线段树应包含4个域,分别是这个区间最长连续的房间数msum,这个区间从左边起的空房间数lsum,这个区间从右边起的空房间数rsum,和标记该房间是否有人mark,然后更新的时候注意一个区间的rsum和相邻区间的lsum是想连的,所以向上更新时可能最长空房间数是由它们两个相加得到的,即区间合并。
代码如下:
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 50005
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
int msum[N<<2],lsum[N<<2],rsum[N<<2],mark[N<<2];
void pushup(int rt,int m)
{
lsum[rt]=lsum[rt<<1]==m-(m>>1)?(lsum[rt<<1]+lsum[rt<<1|1]):(lsum[rt<<1]);
rsum[rt]=rsum[rt<<1|1]==(m>>1)?(rsum[rt<<1|1]+rsum[rt<<1]):(rsum[rt<<1|1]);
msum[rt]=max(rsum[rt<<1]+lsum[rt<<1|1],max(msum[rt<<1],msum[rt<<1|1]));
}
void pushdown(int rt,int m)
{
if(mark[rt]>=0)
{
mark[rt<<1]=mark[rt<<1|1]=mark[rt];
lsum[rt<<1]=rsum[rt<<1]=msum[rt<<1]=mark[rt]?0:(m-(m>>1));
lsum[rt<<1|1]=rsum[rt<<1|1]=msum[rt<<1|1]=mark[rt]?0:(m>>1);
mark[rt]=-1;
}
}
void build(int l,int r,int rt)
{
lsum[rt]=rsum[rt]=msum[rt]=r-l+1;
mark[rt]=-1;
if(l==r) return;
int m=(l+r)>>1;
build(lson);
build(rson);
}
void update(int L,int R,int c,int l,int r,int rt)
{
if(L<=l&&R>=r)
{
msum[rt]=lsum[rt]=rsum[rt]=c?0:r-l+1;
mark[rt]=c;
return;
}
pushdown(rt,r-l+1);
int m=(l+r)>>1;
if(L<=m) update(L,R,c,lson);
if(R>m) update(L,R,c,rson);
pushup(rt,r-l+1);
}
int query(int x,int l,int r,int rt)
{
if(l==r) return l;
pushdown(rt,r-l+1);
int m=(l+r)>>1;
if(x<=msum[rt<<1]) return query(x,lson);
else if(x<=rsum[rt<<1]+lsum[rt<<1|1]) return m-rsum[rt<<1]+1;
else return query(x,rson);
pushup(rt,r-l+1);
}
int main()
{
int n,m;
while(~scanf("%d%d",&n,&m))
{
build(1,n,1);
for(int i=0;i<m;i++)
{
int op,a,b;
scanf("%d",&op);
if(op==1)
{
scanf("%d",&a);
if(msum[1]<a) printf("0\n");
else
{
int pos=query(a,1,n,1);
printf("%d\n",pos);
update(pos,pos+a-1,1,1,n,1);
}
}
else
{
scanf("%d%d",&a,&b);
update(a,a+b-1,0,1,n,1);
}
}
}
return 0;
}