BZOJ1593 [Usaco2008 Feb]Hotel 旅館 線段樹

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

**題解：
用線段樹維護每個區間的最大連續0的個數，當區間進行合併時，要另維護區間左連續零和右連續零的個數，再進行相加。入住操作相當於找第一個區間連續零個數大於k的，再進行區間賦值爲1，退房操作相當於區間賦值爲0。**

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int MAXN=50001;
struct Seg
{
    int l,r,s,lazy,lx,rx,f;
}Tree[MAXN<<2];
int n;
void Make_Tree(int x,int l,int r)
{
    Tree[x].l=l,Tree[x].r=r;
    if(l==r) {
        Tree[x].s=1;Tree[x].lazy=0;Tree[x].lx=Tree[x].rx=1;
        Tree[x].f=1;
        return;
    }
    int mid=(l+r)>>1;
    Make_Tree(x<<1,l,mid);
    Make_Tree(x<<1|1,mid+1,r);
    Tree[x].s=Tree[x].r-Tree[x].l+1;
    Tree[x].lx=Tree[x].rx=Tree[x].s;
} 
void pushdown(int x)
{
    int lc=x<<1,rc=x<<1|1;
    if(Tree[x].f==1) {
        Tree[x<<1].s=Tree[x<<1].lx=Tree[x<<1].rx=Tree[x<<1].r-Tree[x<<1].l+1;
        Tree[x<<1|1].s=Tree[x<<1|1].lx=Tree[x<<1|1].rx=Tree[x<<1|1].r-Tree[x<<1|1].l+1;
        Tree[x].f=0;Tree[x<<1].f=1,Tree[x<<1|1].f=1;
    }
    else if(Tree[x].f==2) {
        Tree[x<<1].s=Tree[x<<1|1].s=0;
        Tree[x<<1].lx=Tree[x<<1|1].lx=0;
        Tree[x<<1].rx=Tree[x<<1|1].rx=0;
        Tree[x].f=0;Tree[x<<1].f=2,Tree[x<<1|1].f=2;
    }
}
void Modify(int x,int l,int r,int f)
{
    //cout<<x<<' '<<l<<' '<<r<<Tree[x].s<<endl;
    if(Tree[x].l==l&&Tree[x].r==r) {
        if(f==1) Tree[x].s=Tree[x].lx=Tree[x].rx=r-l+1,Tree[x].f=1;
        else if(f==2) Tree[x].s=Tree[x].lx=Tree[x].rx=0,Tree[x].f=2;
        return;
    }
    int mid=(Tree[x].l+Tree[x].r)>>1;
    pushdown(x);
    if(r<=mid) Modify(x<<1,l,r,f);
    else if(l>mid) Modify(x<<1|1,l,r,f);
    else Modify(x<<1,l,mid,f),Modify(x<<1|1,mid+1,r,f);
    Tree[x].s=max(Tree[x<<1].s,Tree[x<<1|1].s);
    Tree[x].s=max(Tree[x].s,Tree[x<<1].rx+Tree[x<<1|1].lx);
    Tree[x].lx=Tree[x<<1].lx;
    if(Tree[x].lx==Tree[x<<1].r-Tree[x<<1].l+1) Tree[x].lx+=Tree[x<<1|1].lx;
    Tree[x].rx=Tree[x<<1|1].rx;
    if(Tree[x].rx==Tree[x<<1|1].r-Tree[x<<1|1].l+1) Tree[x].rx+=Tree[x<<1].rx;
  //  cout<<'@'<<Tree[x].s<<' '<<x<<'@'<<endl;
}
int Query(int x,int k)
{
    if(Tree[x].l==Tree[x].r) return Tree[x].l;
    pushdown(x);
    int mid=(Tree[x].l+Tree[x].r)>>1;
    if(Tree[x<<1].s>=k) return Query(x<<1,k);
    else if(Tree[x<<1].rx+Tree[x<<1|1].lx>=k) return mid-Tree[x<<1].rx+1;
    else return Query(x<<1|1,k);
}
void check()
{
    int i,j;
    for(i=1;i<=n*2;i++)
    cout<<Tree[i].l<<" "<<Tree[i].r<<" "<<Tree[i].s<<" "<<Tree[i].lx<<" "<<Tree[i].rx<<" "<<Tree[i].f<<endl;
}
int main(int argc, char *argv[])
{
    int i,j,op,x,y,m;
    scanf("%d%d",&n,&m);
    Make_Tree(1,1,n);
    for(i=1;i<=m;i++)
    {
        scanf("%d",&op);
        if(op==1) {
            scanf("%d",&x);
            if(Tree[1].s<x) printf("0\n");
            else {
                int ans=Query(1,x);
                printf("%d\n",ans);
                Modify(1,ans,ans+x-1,2);
            }
        }
        else {
            scanf("%d%d",&x,&y);
            Modify(1,x,x+y-1,1);
        }
    }
    return 0;
}