题意:给一个n和m,表示n个房间,m次操作,操作类型有2种,一种把求连续未租出的房间数有d个的最小的最左边的房间号,另一个操作时把从x到x+d-1的房间号收回。
分析:这是一个区间合并的典型应用,基本只要套套模板就好了
下面是对区间合并的解释:
up函数:当修改完此时的区间时,为了维护该区间上面区间值的正确性,向上回朔更新上面区间的值
down函数:把延迟标记下传,并修改下面区间的值,维护下面区间的正确性区间
(需要注意的是,up函数只有需要更新的时候才需要用到,而down只要是带延迟标记操作时即使只是遍历也是要加上的)
update函数:在两种操作结束后对线段树的值维护
query函数:求最小的满足条件的房间号
此外对于此题的一些解释:
lsum数组表示该段rt区间的左连续区间的最大值
rsum数组表示该段rt区间的右连续区间的最大值
msum数组表示该段区间的连续区间的最大值,可以是左连续、右连续、也可以使中间连续
lazy数组表示懒惰标记,-1时表示未打上懒惰标记,不需要改变,0时表示该区间打上了区间被使用,1表示该区间被回收
#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
#include<vector>
#pragma comment(linker,"/STACK:1024000000,1024000000")
using namespace std;
const int maxn = 5e4+5;
int msum[maxn<<2],lsum[maxn<<2],rsum[maxn<<2];
int lazy[maxn<<2];
void up(int l,int r,int rt){
int mid=(l+r)>>1;
lsum[rt]=lsum[rt<<1];
rsum[rt]=rsum[rt<<1|1];
if(lsum[rt<<1]==mid-l+1) lsum[rt]+=lsum[rt<<1|1];
if(rsum[rt<<1|1]==r-mid) rsum[rt]+=rsum[rt<<1];
msum[rt]=max(rsum[rt<<1]+lsum[rt<<1|1],max(msum[rt<<1],msum[rt<<1|1]));
}
void down(int l,int r,int rt){
if(lazy[rt]!=-1){
int mid=(l+r)>>1;
lazy[rt<<1]=lazy[rt<<1|1]=lazy[rt];
msum[rt<<1]=lsum[rt<<1]=rsum[rt<<1]=lazy[rt]?mid-l+1:0;
msum[rt<<1|1]=lsum[rt<<1|1]=rsum[rt<<1|1]=lazy[rt]?r-mid:0;
lazy[rt]=-1;
}
}
void build(int l,int r,int rt){
lazy[rt]=-1;
msum[rt]=lsum[rt]=rsum[rt]=r-l+1;
if(l==r) return;
int mid=(l+r)>>1;
build(l,mid,rt<<1);
build(mid+1,r,rt<<1|1);
}
void update(int L,int R,int k,int l,int r,int rt){
if(L<=l&&r<=R){
lazy[rt]=k;
msum[rt]=lsum[rt]=rsum[rt]=k?r-l+1:0;
return;
}
down(l,r,rt);
int mid=(l+r)>>1;
if(L<=mid) update(L,R,k,l,mid,rt<<1);
if(R>mid)update(L,R,k,mid+1,r,rt<<1|1);
up(l,r,rt);
}
int query(int len,int l,int r,int rt){
if(l==r) return l;
down(l,r,rt);
int mid=(l+r)>>1;
if(msum[rt<<1]>=len) return query(len,l,mid,rt<<1);
else if(rsum[rt<<1]+lsum[rt<<1|1]>=len) return mid-rsum[rt<<1]+1;
else return query(len,mid+1,r,rt<<1|1);
}
int main()
{
int n,m;
scanf("%d%d",&n,&m);
build(1,n,1);
while(m--){
int k,x,d;
scanf("%d",&k);
if(k==1){
scanf("%d",&d);
if(d>msum[1]) printf("0\n");
else{
int L=query(d,1,n,1);
printf("%d\n",L);
update(L,L+d-1,0,1,n,1);
}
}
else {
scanf("%d%d",&x,&d);
update(x,x+d-1,1,1,n,1);
}
}
}