【题】【线段树(区间·LAZY)】NKOJ 1550 旅馆 【Usaco Feb08 Gold】

NKOJ 1550 旅馆 【Usaco Feb08 Gold】
时间限制 : 10000 MS 空间限制 : 65536 KB

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

来源 usaco feb2008 金组

思路、
“连续”说明是连续区间问题
“左下”可看做给一段区间涂色，“离开”看做擦除
“编号最小”解决方法是查找时先查左，再查中间，最后右边

#include<cstdio>
#include<iostream>
using namespace std;
const int need=50005;

struct fy{int a,b,lazy,fl,fr,fmax;}t[need*2];//lazy=1表示住下 ==-1表示离开

int le[need*2],ri[need*2],tot=0;
int k,x,y;

void build(int x,int y)
{
    int s=++tot;
    t[s].a=x,t[s].b=y;
    t[s].fl=t[s].fr=t[s].fmax=y-x+1; 
    if(x==y) return;
    le[s]=tot+1;build(x,(x+y)>>1);
    ri[s]=tot+1;build((x+y)/2+1,y);
}

void putdown(int s)
{
    int k=t[s].lazy;
    t[s].lazy=0;
    t[le[s]].lazy=t[ri[s]].lazy=k;
    if(k==1)//左下后整个区间无连续空位，所以全为0
    {
        t[le[s]].fl=t[le[s]].fr=t[le[s]].fmax=0;
        t[ri[s]].fl=t[ri[s]].fr=t[ri[s]].fmax=0;
    }
    else //离开后整个区间都是空位，所以最大值都为区间长度
    {
        t[le[s]].fl=t[le[s]].fr=t[le[s]].fmax=t[le[s]].b-t[le[s]].a+1;
        t[ri[s]].fl=t[ri[s]].fr=t[ri[s]].fmax=t[ri[s]].b-t[ri[s]].a+1;
    }
} 

int find(int s)//返回最靠左的区间的左方编号
{
    if(t[s].fmax<k) return 0;//返回0表示找不到，若没返回则一定能找到
    if(t[s].lazy!=0) putdown(s);
    if(t[le[s]].fmax>=k) return find(le[s]);
    if(t[ri[s]].fl+t[le[s]].fr>=k) return t[le[s]].b-t[le[s]].fr+1;
    else return find(ri[s]);
    //先查左，再查中间，最后右边
}

int max3(int a,int b,int c)
{
    return max(a,max(b,c));
}

void NBHB(int s)// 更新节点的各值
{
    if(t[s].a==t[s].b) return ;
    int lenl=t[le[s]].b-t[le[s]].a+1,lenr=t[ri[s]].b-t[ri[s]].a+1;
    t[s].fl=lenl==t[le[s]].fl?lenl+t[ri[s]].fl:t[le[s]].fl;
    t[s].fr=lenr==t[ri[s]].fr?lenr+t[le[s]].fr:t[ri[s]].fr;
    t[s].fmax=max3(t[le[s]].fmax,t[ri[s]].fmax,t[le[s]].fr+t[ri[s]].fl);
}

void change(int s,int ma)
{
    if(t[s].a>y||t[s].b<x) return ;
    if(x<=t[s].a&&t[s].b<=y)
    {
        if(ma==1) t[s].fl=t[s].fr=t[s].fmax=0;
        else t[s].fl=t[s].fr=t[s].fmax=t[s].b-t[s].a+1;
        //不能直接NBHB，不然若s为叶节点就会出错，或者在NBHB中特判一下
        t[s].lazy=ma;
        putdown(s);
        return ;
    } 
    if(t[s].lazy!=0) putdown(s);
    change(le[s],ma),change(ri[s],ma);
    NBHB(s);
}

int main()
{
    int n,m;scanf("%d%d",&n,&m);
    if(n==1)//特判不能建树的情况！！！！！ 
    {
        bool use=false;
        for(int i=1,a;i<=m;i++)
        {
            scanf("%d",&a);
            if(a==1)
            {
                scanf("%d",&k);
                if(k==1&&!use) puts("1");
                else puts("0");
            }
            else if(a==2)
            {
                scanf("%d%d",&x,&k);
                if(x==1&&k==1) use=false;
            }
        }
        return 0;
    }
    build(1,n);
    for(int i=1,a,b;i<=m;i++)
    {
        scanf("%d",&a);
        if(a==1)
        {
            scanf("%d",&k);
            b=find(1);
            printf("%d\n",b);
            if(b!=0) 
            {
                x=b,y=x+k-1;
                change(1,1);
            }
        }
        else
        {
            scanf("%d%d",&x,&k);
            y=x+k-1;
            if(y>n) y=n;//万一超了呢 
            change(1,-1);
        }
    }
}