[USACO08FEB]酒店Hotel

题目大意

一段01序列，0代表没人住，1代表有人住,住房和退房都要求是修改一段连续区间.

求能够安排住房的区间的左端点

题解

线段树

要判断能否住 要维护区间0序列长最大值.和区间0序列最左边的长度和右边长度

代码

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

int sum[400010],lm[400010],rm[400010],m[400010],tag[400010],n,T;

void pushup(int rt){
    if(m[rt<<1]==sum[rt<<1]){
        lm[rt]=m[rt<<1]+lm[rt<<1|1];
    }else{
        lm[rt]=lm[rt<<1];
    }
    if(m[rt<<1|1]==sum[rt<<1|1]){
        rm[rt]=m[rt<<1|1]+rm[rt<<1];
    }else{
        rm[rt]=rm[rt<<1|1];
    }
    m[rt]=max(max(m[rt<<1],m[rt<<1|1]),rm[rt<<1]+lm[rt<<1|1]);
    return;
}

void pushdown(int rt){
    if(tag[rt]==0)    return;
    if(tag[rt]==1){
        tag[rt<<1]=tag[rt<<1|1]=1;
        m[rt<<1]=lm[rt<<1]=rm[rt<<1]=0;
        m[rt<<1|1]=lm[rt<<1|1]=rm[rt<<1|1]=0;
    }
    if(tag[rt]==2){
        tag[rt<<1]=tag[rt<<1|1]=2;
        m[rt<<1]=lm[rt<<1]=rm[rt<<1]=sum[rt<<1];
        m[rt<<1|1]=lm[rt<<1|1]=rm[rt<<1|1]=sum[rt<<1|1];
    }
    tag[rt]=0;
}

void build(int rt,int l,int r){
    lm[rt]=rm[rt]=m[rt]=sum[rt]=r-l+1;
    tag[rt]=0;
    if(l==r) return;
    int mid=(l+r)>>1;
    build(rt<<1,l,mid);
    build(rt<<1|1,mid+1,r);
    return;
}

int query(int rt,int l,int r,int len){
    pushdown(rt);
    if(l==r)    return l;
    int mid=(l+r)>>1;
    if(m[rt<<1]>=len)    return query(rt<<1,l,mid,len);
    if(rm[rt<<1]+lm[rt<<1|1]>=len)    return mid-rm[rt<<1]+1;
    else    return query(rt<<1|1,mid+1,r,len);
}

void modify(int L,int R,int c,int rt,int l,int r){
    pushdown(rt);
    if((L<=l)&&(r<=R)){
        if(c==1) m[rt]=lm[rt]=rm[rt]=0;
        else m[rt]=lm[rt]=rm[rt]=sum[rt];
        tag[rt]=c;
        return;
    }
    int mid=(l+r)>>1;
    if(L<=mid)    modify(L,R,c,rt<<1,l,mid);
    if(R>mid)    modify(L,R,c,rt<<1|1,mid+1,r);
    pushup(rt);
}

int main(){
    scanf("%d%d",&n,&T);
    build(1,1,n);
    while(T--){
        int opt,a,b;
        scanf("%d",&opt);
        if(opt==1){
            scanf("%d",&a);
            if(m[1]<a){printf("0\n");continue;}
            int p=query(1,1,n,a);
            printf("%d\n",p);
            modify(p,p+a-1,1,1,1,n);
        }else{
            scanf("%d%d",&a,&b);
            modify(a,a+b-1,2,1,1,n);
        }
    }
    return 0;
}