BZOJ 1593 [Usaco2008 Feb]Hotel 旅馆 双倍经验,线段树
1593: [Usaco2008 Feb]Hotel 旅馆
做完本题可以去这道题看看 BZOJ3050 Seating
Description
奶牛们最近的旅游计划,是到苏必利尔湖畔,享受那里的湖光山色,以及明媚的阳光。作为整个旅游的策划者和负
责人,贝茜选择在湖边的一家著名的旅馆住宿。这个巨大的旅馆一共有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)个按输入次序到来的住店或退房的请求。第一个请求到来前,旅店中所有房间都是空闲的。
Input
* 第1行: 2个用空格隔开的整数:N、M
* 第2..M+1行: 第i+1描述了第i个请求,
如果它是一个订房请求,则用2个数字 1、D_i描述,数字间用空格隔开;
如果它是一个退房请求,用3 个以空格隔开的数字2、X_i、D_i描述
Output
* 第1..??行: 对于每个订房请求,输出1个独占1行的数字:
如果请求能被满足 ,输出满足条件的最小的r;如果请求无法被满足,输出0
Sample Input
10 6
1 3
1 3
1 3
1 3
2 5 5
1 6
1 3
1 3
1 3
1 3
2 5 5
1 6
Sample Output
1
4
7
0
5
4
7
0
5
思路:
线段树区间合并裸题 我们设这样几个数组 : siz、lsiz、rsiz、len、laz来分别表示区间最大连续长度,从左开始的最大连续长度,从右开始的区间最大连续长度、区间总长度、懒度标记
这样 我们的siz就是两个儿子的siz和左儿子的rsiz+右儿子的lsiz 对于lsiz的维护,若左儿子的siz等于len,那么就是左儿子siz加右儿子lsiz,否则是左儿子的lsiz,rsiz类似。
本题输入有小坑,退房的x、y是从x到x+y-1.
至于lazytag,不会的同学请去学习一下区间修改线段树的lazy用法。change部分只有这一小点
对于每次查询,如果siz[root] < query ,那么直接输出0即可,否则的话以 左子树? 左子树右加右子树左 ? 右子树的顺序选择递归,如果左子树不够大,而中间部分足够,那么我们可以直接返回ans并从ans+query的区间范围对树进行修改。详细的内容请看代码 如下:
#include <cstdio> #include <cstring> #include <algorithm> #include <iostream> #define ls p<<1 #define rs p<<1|1 #define lson l,m,p<<1 #define rson m+1,r,p<<1|1 #define im int mid = (l+r) >> 1 using namespace std; const int N = 410000; int ans=0,n; struct SegTree { int lsz[N<<2], rsz[N<<2], siz[N<<2], laz[N<<2], len[N<<2]; void pushup(int p) { lsz[p]=(siz[ls]==len[ls]?len[ls]+lsz[rs]:lsz[ls]); rsz[p]=(siz[rs]==len[rs]?len[rs]+rsz[ls]:rsz[rs]); siz[p]=max(siz[ls],max(siz[rs],rsz[ls]+lsz[rs])); } void pushdown(int p) { if(laz[p]==1) { lsz[rs]=rsz[rs]=siz[rs]=lsz[ls]=rsz[ls]=siz[ls]=0; laz[ls]=laz[rs]=1; } else if(laz[p]==2) { lsz[rs]=rsz[rs]=siz[rs]=len[rs]; lsz[ls]=rsz[ls]=siz[ls]=len[ls]; laz[ls]=laz[rs]=2; } laz[p]=0; } void build(int l,int r,int p) { lsz[p]=rsz[p]=siz[p]=len[p]=r-l+1; if(l==r)return; im; build(l,mid,ls); build(mid+1,r,rs); } int query(int l,int r,int p,int x) { pushdown(p); if (l==r) return l; int m = (l+r)>>1; if (siz[ls]>=x) return query(lson,x); if (rsz[ls]+lsz[ls|1]>=x) return m-rsz[ls]+1; else return query(rson,x); } void change(int l,int r,int p,int x,int y,int c) { pushdown(p); if(x<=l&&y>=r) { if(c==1) siz[p]=lsz[p]=rsz[p]=0; else siz[p]=lsz[p]=rsz[p]=len[p]; laz[p]=c; return; } im; if(x<=mid) change(l,mid,ls,x,y,c); if(y>mid) change(mid+1,r,rs,x,y,c); pushup(p); } }Tr; int main() { int q; scanf("%d%d",&n,&q); Tr.build(1,n,1); while (q--) { int opt,x,y,p; scanf("%d",&opt); if (opt==1) { scanf("%d",&x); if (Tr.siz[1]<x) { printf("0\n"); continue; } p = Tr.query(1,n,1,x); printf("%d\n",p); Tr.change(1,n,1,p,p+x-1,1); } else { scanf("%d%d",&x,&y); Tr.change(1,n,1,x,x+y-1,2); } } }
欢迎来原博客看看 >原文链接<