[Usaco2008 Feb][BZOJ1593] Hotel 旅馆|线段树

1593: [Usaco2008 Feb]Hotel 旅馆

Time Limit: 10 Sec Memory Limit: 64 MB
Submit: 417 Solved: 243
[Submit][Status][Discuss]

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

HINT

Source

Gold

ccccccccccccccccccccccccc………………

一开始ask操作忘记pushdown了，调了一个小时。

裸线段树啊，没什么好说的，注意合并。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#define N 50005
using namespace std;
int n,m;
int lc[4*N],rc[4*N],mx[4*N],len[4*N],l[4*N],r[4*N],tag[4*N];
inline int read()
{
    int a=0,f=1; char c=getchar();
    while (c<'0'||c>'9') {if (c=='0') f=-1; c=getchar();}
    while (c>='0'&&c<='9') {a=a*10+c-'0'; c=getchar();}
    return a*f;
}
inline void pushup(int k)
{
    if (mx[k<<1]==len[k<<1]&&mx[k<<1|1]==len[k<<1|1]) lc[k]=rc[k]=mx[k]=len[k];
    else if (mx[k<<1]==len[k<<1]) {lc[k]=mx[k]=len[k<<1]+lc[k<<1|1]; rc[k]=rc[k<<1|1];}
    else if (mx[k<<1|1]==len[k<<1|1]) {rc[k]=mx[k]=len[k<<1|1]+rc[k<<1]; lc[k]=lc[k<<1];}
    else
    {
        lc[k]=lc[k<<1]; rc[k]=rc[k<<1|1];
        mx[k]=max(mx[k<<1],mx[k<<1|1]);
        mx[k]=max(mx[k],rc[k<<1]+lc[k<<1|1]);
    }
}
inline void pushdown(int k)
{
    if (tag[k]==-1||l[k]==r[k]) return;
    tag[k<<1]=tag[k<<1|1]=tag[k];
    lc[k<<1]=rc[k<<1]=mx[k<<1]=(tag[k]==0?len[k<<1]:0);
    lc[k<<1|1]=rc[k<<1|1]=mx[k<<1|1]=(tag[k]==0?len[k<<1|1]:0);
    tag[k]=-1;
}
void change(int k,int x,int y,int z)
{
    pushdown(k);
    if (l[k]==x&&r[k]==y)
    {
        lc[k]=rc[k]=mx[k]=(z==0?len[k]:0);
        tag[k]=z;
        return;
    }
    int mid=(l[k]+r[k])>>1;
    if (y<=mid) change(k<<1,x,y,z);
    else if (mid<x) change(k<<1|1,x,y,z);
    else
    {
        change(k<<1,x,mid,z);
        change(k<<1|1,mid+1,y,z);
    }
    pushup(k);
}
void build(int k,int x,int y)
{
    l[k]=x; r[k]=y; lc[k]=rc[k]=mx[k]=len[k]=y-x+1; tag[k]=-1;
    if (x==y) return;
    int mid=(l[k]+r[k])>>1;
    build(k<<1,x,mid); build(k<<1|1,mid+1,y);
}
inline int ask(int k,int d)
{
    pushdown(k);
    if (l[k]==r[k])return l[k];
    int mid=(l[k]+r[k])>>1;
    if (mx[k<<1]>=d) return ask(k<<1,d);
       if (rc[k<<1]+lc[k<<1|1]>=d)
        return mid-rc[k<<1]+1;
    return ask(k<<1|1,d);    
}
int main()
{
    n=read(); m=read();
    build(1,1,n);
    while (m--)
    {
        int tmp=read(),x,d,t;
        if (tmp==1) {d=read(); if (mx[1]<d) puts("0"); else {t=ask(1,d); printf("%d
",t); change(1,t,t+d-1,1);}}
        if (tmp==2) {x=read(); d=read(); change(1,x,x+d-1,0);}
    }
    return 0;
}