pku1823 Hotel

线段树+模拟

空房间标记为1，已入住房间标记为0，这种表示便于统计某段区间内的连续空段长度

 

 

#include <iostream>
#include <algorithm>
using namespace std;

#define MAXN 160010
#define Max(a,b) (a>b?a:b)

int n,p;

struct Node{
    int l,r,
    mx,//最大空段长度
    lmx,//从区间左端开始的空段的最大长度
    rmx;//

    bool flag1,//true表示该段全为1（即该段全为空）
    flag0;//true表示该段全为0（该段无空位）
}nod[3*MAXN];


void build(int u,int l,int r){
    nod[u].flag1=true;
    nod[u].flag0=false;
    nod[u].l=l;
    nod[u].r=r;
    nod[u].lmx=nod[u].rmx=nod[u].mx=r-l+1;
    if(l==r)
        return;
    build(2*u,l,(l+r)/2);
    build(2*u+1,(l+r)/2+1,r);
}

void checkin(int u,int l,int r){
    // 将父亲信息继承给左右儿子//*
    if(nod[u].flag1){
        nod[u].flag1=false;

        nod[2*u].flag1=true;
        nod[2*u].flag0=false;
        nod[2*u].mx=nod[2*u].lmx=nod[2*u].rmx=nod[2*u].r-nod[2*u].l+1;
        

        nod[2*u+1].flag1=true;
        nod[2*u+1].flag0=false;
        nod[2*u+1].mx=nod[2*u+1].lmx=nod[2*u+1].rmx=nod[2*u+1].r-nod[2*u+1].l+1;
        
    }
    
    if(l<=nod[u].l && nod[u].r<=r){    
        nod[u].lmx=nod[u].rmx=nod[u].mx=0;
        nod[u].flag0=true;
        return;
    }
    if(l<=nod[2*u].r)
        checkin(2*u,l,r);
    if(nod[2*u+1].l<=r)
        checkin(2*u+1,l,r);

    // 由左右儿子更新维护父亲//*
    nod[u].mx=Max(nod[2*u].mx,nod[2*u+1].mx);
    nod[u].mx=Max(nod[u].mx,nod[2*u].rmx+nod[2*u+1].lmx);

    if(nod[2*u].lmx==nod[2*u].r-nod[2*u].l+1)
        nod[u].lmx=nod[2*u].lmx+nod[2*u+1].lmx;
    else
        nod[u].lmx=nod[2*u].lmx;


    if(nod[2*u+1].rmx==nod[2*u+1].r-nod[2*u+1].l+1)
        nod[u].rmx=nod[2*u+1].rmx+nod[2*u].rmx;
    else
        nod[u].rmx=nod[2*u+1].rmx;

    if(nod[u].mx==0)
        nod[u].flag0=true;
}

void checkout(int u,int l,int r){
    if(nod[u].flag0){
        nod[u].flag0=false;

        nod[2*u].flag0=true;
        nod[2*u].flag1=false;
        nod[2*u].mx=nod[2*u].lmx=nod[2*u].rmx=0;
        
    
        nod[2*u+1].flag0=true;
        nod[2*u+1].flag1=false;
        nod[2*u+1].mx=nod[2*u+1].lmx=nod[2*u+1].rmx=0;
        
    }

    if(l<=nod[u].l && nod[u].r<=r){    
        nod[u].lmx=nod[u].rmx=nod[u].mx=nod[u].r-nod[u].l+1;
        nod[u].flag1=true;
        return;
    }
    if(l<=nod[2*u].r)
        checkout(2*u,l,r);
    if(nod[2*u+1].l<=r)
        checkout(2*u+1,l,r);

    nod[u].mx=Max(nod[2*u].mx,nod[2*u+1].mx);
    nod[u].mx=Max(nod[u].mx,nod[2*u].rmx+nod[2*u+1].lmx);

    if(nod[2*u].lmx==nod[2*u].r-nod[2*u].l+1)
        nod[u].lmx=nod[2*u].lmx+nod[2*u+1].lmx;
    else
        nod[u].lmx=nod[2*u].lmx;


    if(nod[2*u+1].rmx==nod[2*u+1].r-nod[2*u+1].l+1)
        nod[u].rmx=nod[2*u+1].rmx+nod[2*u].rmx;
    else
        nod[u].rmx=nod[2*u+1].rmx;


    if(nod[u].mx==nod[u].r-nod[u].l+1)
        nod[u].flag1=true;
}

int main(){
    int c,l,r;
    while(scanf("%d%d",&n,&p)!=EOF){
        build(1,1,n);
        while(p--){
            scanf("%d",&c);
            if(c==1){
                scanf("%d%d",&l,&r);
                checkin(1,l,l+r-1);
            }
            else if(c==2){
                scanf("%d%d",&l,&r);
                checkout(1,l,l+r-1);
            }
            else
                printf("%d\n",nod[1].mx);
        }
    }
    return 0;
}