【洛谷P2894】Hotel 线段树+二分查询

题目大意：给定一个长度为 N 的序列，每个点有两种状态 1/0，表示占有和空闲，现支持 first-fit 查询是否有一段连续的长度为 X 的空闲子序列和区间赋值操作。

题解：get到了线段树新技能。。qwq

区间赋值操作和其他线段树一样，维护标记即可。

查询是否有一段连续的长度为 X 的空闲子序列只需要类似于维护区间连续最长字段即可，即：lmx,rmx,mx。first-fit 查询操作需要类似于在线段树上二分的操作，即：若左子树满足条件就走左子树，左右子树能拼接成满足需求的结果则返回，否则遍历右子树。

代码如下

#include <bits/stdc++.h>
using namespace std;
const int maxn=5e4+10;

int n,m,opt;
struct node{
	#define ls(o) t[o].lc
	#define rs(o) t[o].rc
	int lc,rc;
	int mx,lmx,rmx,tag; // 1 -> all zero    2 -> all one
}t[maxn<<1];
int tot,root;
inline void pushup(int o,int l,int r){
	int mid=l+r>>1;
	t[o].lmx=t[ls(o)].lmx==mid-l+1?t[ls(o)].lmx+t[rs(o)].lmx:t[ls(o)].lmx;
	t[o].rmx=t[rs(o)].rmx==r-mid?t[ls(o)].rmx+t[rs(o)].rmx:t[rs(o)].rmx;
	t[o].mx=max(t[ls(o)].rmx+t[rs(o)].lmx,max(t[ls(o)].mx,t[rs(o)].mx));
}
inline void pushdown(int o,int l,int r){
	if(!t[o].tag)return;
	int mid=l+r>>1;
	t[ls(o)].tag=t[rs(o)].tag=t[o].tag;
	t[ls(o)].lmx=t[ls(o)].rmx=t[ls(o)].mx=t[o].tag==1?mid-l+1:0;
	t[rs(o)].lmx=t[rs(o)].rmx=t[rs(o)].mx=t[o].tag==1?r-mid:0;
	t[o].tag=0;
}
int build(int l,int r){
	int o=++tot;
	if(l==r){t[o].lmx=t[o].rmx=t[o].mx=1;return o;}
	int mid=l+r>>1;
	ls(o)=build(l,mid),rs(o)=build(mid+1,r);
	pushup(o,l,r);
	return o;
}
void modify(int o,int l,int r,int x,int y,int flag){
	if(l==x&&r==y){
		t[o].lmx=t[o].rmx=t[o].mx=flag==1?r-l+1:0;
		t[o].tag=flag;
		return;
	}
	int mid=l+r>>1;
	pushdown(o,l,r);
	if(y<=mid)modify(ls(o),l,mid,x,y,flag);
	else if(x>mid)modify(rs(o),mid+1,r,x,y,flag);
	else modify(ls(o),l,mid,x,mid,flag),modify(rs(o),mid+1,r,mid+1,y,flag);
	pushup(o,l,r);
}
int query(int o,int l,int r,int len){
	if(l==r)return l;
	int mid=l+r>>1;
	pushdown(o,l,r);
	if(len<=t[ls(o)].mx)return query(ls(o),l,mid,len);
	else if(len<=t[ls(o)].rmx+t[rs(o)].lmx)return mid-t[ls(o)].rmx+1;
	else return query(rs(o),mid+1,r,len);
}

int main(){
	scanf("%d%d",&n,&m);
	root=build(1,n);
	while(m--){
		scanf("%d",&opt);
		if(opt==1){
			int x;
			scanf("%d",&x);
			if(t[root].mx<x)puts("0");
			else{
				int l=query(root,1,n,x);
				modify(root,1,n,l,l+x-1,2);
				printf("%d\n",l);
			}
		}else if(opt==2){
			int x,y;
			scanf("%d%d",&x,&y);
			modify(root,1,n,x,x+y-1,1);
		}
	}
	return 0;
}