Hotel——线段树

Hotel
题目描述

奶牛们最近的旅游计划，是到苏必利尔湖畔，享受那里的湖光山色，以及明媚的阳光。作为整个旅游的策划者和负责人，贝茜选择在湖边的一家著名的旅馆住宿。这个巨大的旅馆一共有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

样例输入

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

样例输出
1
4
7
0
5

---

一开始呢，是不会做这道题的。。。对于在一个区间内“缝缝补补”（染色）之后来找连续的一个长度限定的（len>=d）这样一个区间，是比较陌生的。

然后呢，可爱的ljh大神就描述了一下他的做法，对于每个节点，有lmax,rmax,maxx三个域，分别表示当前区间左边连续的最长子区间，右边开始连续的最长子区间，以及整个区间中的最长子区间。

这样一想，就知道怎么insert了，关键就在要把当前区间i的i*2和i*2+1两个子区间的中间连起来：tree[i*2].rmax+tree[i*2+1].lmax
还有用tree[i].hotel做一个lazy-tag，延后染色，一切都那么和谐。

但是在寻找x（最近的连续长度超过d的区间的左端点）时，挂了。
详细的呢，看注释吧。

#include<cstdio>
#include<algorithm>
using namespace std;
#define MAXN 50000
struct node{ 
    int l,r;
    int lmax,rmax,maxx,pos;
    int hotel;
}tree[MAXN*4+100];
int n,m;
void Max(int x,int y,node &z,int xpos,int ypos)
{
    if(x<y)
        z.maxx=y,z.pos=ypos;
    else
        z.maxx=x,z.pos=xpos;
}
void build(int i,int l,int r)
{
    tree[i].l=l,tree[i].r=r;
    tree[i].lmax=tree[i].rmax=tree[i].maxx=r-l+1;
    tree[i].pos=l;
    if(l==r) return ;
    int mid=(l+r)/2;
    build(i*2,l,mid);
    build(i*2+1,mid+1,r);
}
void cover(node &x,int len,int pos)
{
    x.lmax=x.rmax=x.maxx=len;
    x.pos=pos;
}
void insert(int i,int l,int r,int f)
{
    if(r<tree[i].l || tree[i].r<l) return ;
    if(l<=tree[i].l && tree[i].r<=r){
        tree[i].hotel=f;
        if(f)
            cover(tree[i],0,-1);
        else
            cover(tree[i],tree[i].r-tree[i].l+1,tree[i].l);
        return ; 
    }
    if(tree[i].hotel>=0){ //lazy-tag
        tree[i*2].hotel=tree[i*2+1].hotel=tree[i].hotel;
        if(tree[i].hotel){
            cover(tree[i*2],0,-1);
            cover(tree[i*2+1],0,-1);
        }
        else{
            cover(tree[i*2],tree[i*2].r-tree[i*2].l+1,tree[i*2].l);
            cover(tree[i*2+1],tree[i*2+1].r-tree[i*2+1].l+1,tree[i*2+1].l);
        }
    }
    tree[i].hotel=-1;
    insert(i*2,l,r,f);
    insert(i*2+1,l,r,f);

    int mid=(tree[i].l+tree[i].r)/2;
    tree[i].lmax=tree[i*2].lmax,tree[i].rmax=tree[i*2+1].rmax;
    if(tree[i].lmax==tree[i*2].r-tree[i*2].l+1)
        tree[i].lmax+=tree[i*2+1].lmax;
    if(tree[i].rmax==tree[i*2+1].r-tree[i*2+1].l+1)
        tree[i].rmax+=tree[i*2].rmax;

    Max(tree[i*2].maxx,tree[i*2+1].maxx,tree[i],tree[i*2].pos,tree[i*2+1].pos);
    Max(tree[i].maxx,tree[i*2].rmax+tree[i*2+1].lmax,tree[i],tree[i].pos,mid-tree[i*2].rmax+1);
    Max(tree[i].maxx,tree[i*2].lmax,tree[i],tree[i].pos,tree[i].l);
    Max(tree[i].maxx,tree[i*2+1].rmax,tree[i],tree[i].pos,tree[i].r);
}
int find_room(int i,int d)
{
    if(tree[i].maxx<d) return -1;  //没有任一区间满足长度超过d的空闲区间
    if(tree[i].l==tree[i].r) return tree[i].l;   //important
    if(tree[i].lmax>=d) return tree[i].l;  //都是被迫递归下来的，能有区间左端点就是最好的了
    if(tree[i*2].maxx>=d) return find_room(i*2,d); //左区间又肯定在左区间找
    else if(tree[i*2].rmax+tree[i*2+1].lmax>=d)  //左区间不够，但是中间那一截够了也是不错的，至少比右区间的要近些
        return tree[i*2].r-tree[i*2].rmax+1;
    else return find_room(i*2+1,d); //实在不行了，去右区间。。。
}
int main()
{
    int flag,d,x;
    scanf("%d%d",&n,&m);
    build(1,1,n);
    for(int i=1;i<=m;i++){
        scanf("%d",&flag);
        if(flag==1){
            scanf("%d",&d);
            x=find_room(1,d);
            if(x==-1)
                printf("0\n");
            else{
                printf("%d\n",x);
                insert(1,x,x+d-1,1);
            }
        }
        else{
            scanf("%d%d",&x,&d);
            insert(1,x,x+d-1,0);
        }
    }
}

转载于:https://www.cnblogs.com/katarinayuan/p/6572869.html