题目链接:点击打开链接
题目大意:
原题为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);
}
}
}
}