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

Sample Output

1
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
");
                continue;
            }
            p = Tr.query(1,n,1,x);
            printf("%d
",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);
        }
    }
}

 欢迎来原博客看看 >原文链接<

 

原文地址:https://www.cnblogs.com/Tobichi/p/9079015.html