1016 - 线段树 - 旅馆(BZOJ xxx)

版权声明:虽然我只是个小蒟蒻但转载也请注明出处哦 https://blog.csdn.net/weixin_42557561/article/details/83096194

题目描述

奶牛们最近的旅游计划,是到苏必利尔湖畔,享受那里的湖光山色,以及明媚的阳光。作为整个旅游的策划者和负责人,贝茜选择在湖边的一家著名的旅馆住宿。这个巨大的旅馆一共有 N (1 <= N <= 50,000)间客房,它们在同一层楼中顺次一字排开,在任何一个房间里,只需要拉开窗帘,就能见到波光粼粼的湖面。

贝茜一行,以及其他慕名而来的旅游者,都是一批批地来到旅馆的服务台,希望能订到 D_i (1 <= D_i <= N) 间连续的房间。服务台的接待工作也很简单:如果存在 r 满足编号为 r..r+D_i-1 的房间均空着,他就将这一批顾客安排到这些房间入住;如果没有满足条件的 r,他会道歉说没有足够的空房间,请顾客们另找一家宾馆。如果有多个满足条件的 r,服务员会选择其中最小的一个。

旅馆中的退房服务也是批量进行的。每一个退房请求由 2 个数字 X_i、D_i 描述,表示编号为 X_i..X_i+D_i-1 (1 <= X_i <= N-D_i+1) 房间中的客人全部离开。退房前,请求退掉的房间中的一些,甚至是所有,可能本来就无人入住。

而你的工作,就是写一个程序,帮服务员为旅客安排房间。你的程序一共需要处理 M (1 <= M < 50,000)个按输入次序到来的住店或退房的请求。第一个请求到来前,旅店中所有房间都是空闲的。

输入格式

第 1 行: 2 个用空格隔开的整数:N、M 。
第 2.. M+1行: 第 i+1 描述了第 i 个请求,如果它是一个订房请求,则用 2 个数字 1、D_i描述,数字间用空格隔开;如果它是一个退房请求,用 3 个以空格隔开的数字 2、X_i、D_i 描述.

输出格式

第 1 ... ?? 行: 对于每个订房请求,输出 1 个独占 1 行的数字:如果请求能被满足,输出满足条件的最小的 r;如果请求无法被满足,输出 0 。

样例数据 1

输入 

10 6 
1 3 
1 3 
1 3 
1 3 
2 5 5 
1 6

输出





5

分析

以前都只做过线段树的板子题,现如今碰到这道题,这感觉就跟没学线段树一样……

不过还好,在gsj大佬的带领下,还是搞出来了

订房:查询是否有连续D个位置是空的,如果有输出最左端点

退房:将X到X+D-1这段位置置为空

这样一转述,就很容易想到线段树了,区间查询,区间修改

每个节点存储3个信息:

  1. 从左数最多有多少个连续的0  lmax
  2. 从右数最多有多少个连续的0  rmax
  3. 整个区间最多有多少个连续的0  maxn

 一开始的时候这些的值都赋为区间的长度,因为整个区间都是空的

当遇到一个查询的时候,就从root开始,如果左边的连续0的个数大于我们需要的D,就直接返回 左端点

然后判断中间

最后判断右边

(因为题目要求我们输出最左端的一种情况,每次我们都从左开始判断,就能保证最后得到的答案一定是最靠左的了)

遇到修改的时候,就该怎么修改怎么修改,只是在最后pushup的时候稍微复杂一点点

有几种情况需要判断

最后需要注意:由于我们涉及了区间修改,所以每做一步操作都别忘了pushdown

代码

#include<bits/stdc++.h>
#define in read()
#define lc (k<<1)
#define rc (k<<1)|1//
#define N 50009
using namespace std;
inline int read(){
	char ch;int f=1,res=0;
	while((ch=getchar())<'0'||ch>'9') if(ch=='-') f=-1;
	while(ch>='0'&&ch<='9'){
		res=(res<<3)+(res<<1)+ch-'0';
		ch=getchar();
	}
	return f==1?res:-res;
}
int n,m;
int lmax[4*N],rmax[4*N],maxn[4*N],lzy[4*N];
void build(int k,int l,int r){
	if(l==r){
		lmax[k]=rmax[k]=maxn[k]=1;
		return ;
	}
	int mid=l+r>>1;
	build(lc,l,mid);build(rc,mid+1,r);
	lmax[k]=rmax[k]=maxn[k]=r-l+1;
}
void pushdown(int k,int l,int r){
	int mid=l+r>>1;
	if(!maxn[k]){
		lmax[lc]=rmax[lc]=maxn[lc]=0;
		lmax[rc]=rmax[rc]=maxn[rc]=0;
	}
	else{
		lmax[lc]=rmax[lc]=maxn[lc]=mid-l+1;
		lmax[rc]=rmax[rc]=maxn[rc]=r-mid;
	}
	lzy[k]=0;lzy[lc]=lzy[rc]=1;
}
int query(int k,int l,int r,int x){
	if(l==r) return l;
	int mid=l+r>>1;
	if(lzy[k]) pushdown(k,l,r);
	if(lmax[k]>=x) return l;
	if(maxn[lc]>=x) return query(lc,l,mid,x);
	if(rmax[lc]+lmax[rc]>=x) return mid-rmax[lc]+1;
	return query(rc,mid+1,r,x);
}
void pushup(int k,int l,int r){
	int mid=l+r>>1;
	if(lmax[lc]!=mid-l+1) lmax[k]=lmax[lc];
	else lmax[k]=lmax[lc]+lmax[rc];
	if(rmax[rc]!=r-mid) rmax[k]=rmax[rc];
	else rmax[k]=rmax[lc]+rmax[rc];
	maxn[k]=max(lmax[k],rmax[k]);
	maxn[k]=max(maxn[k],rmax[lc]+lmax[rc]);
	maxn[k]=max(maxn[k],maxn[lc]);maxn[k]=max(maxn[k],maxn[rc]);
}
void modify(int k,int l,int r,int x,int y,int v){
	if(x<=l&&r<=y){
		if(!v){
			lmax[k]=rmax[k]=maxn[k]=r-l+1;
			lzy[k]=1;
		}
		else{
			lmax[k]=rmax[k]=maxn[k]=0;
			lzy[k]=1;
		}
		return;
	}
	if(lzy[k]) pushdown(k,l,r);
	int mid=l+r>>1;
	if(x<=mid) modify(lc,l,mid,x,y,v);
	if(y>mid) modify(rc,mid+1,r,x,y,v);
	pushup(k,l,r);
}
int main(){
	n=in;m=in;
	int i,j,k,ty,x,d;
	build(1,1,n);
	for(i=1;i<=m;++i){
		ty=in;
		if(ty==1){
			d=in;
			if(maxn[1]<d) printf("0\n");
			else{
				int pos=query(1,1,n,d);
				printf("%d\n",pos);
				modify(1,1,n,pos,pos+d-1,1);//1-->be used
			}
		}
		else{
			x=in;d=in;
			modify(1,1,n,x,x+d-1,0);
		}
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_42557561/article/details/83096194
今日推荐