[bzoj1593]旅馆

用线段树维护区间中最大的一段连续的1，以左端点为左端点最大的一段连续的1，以右端点为右端点最大的一段连续的1，然后就可以支持区间修改和查询了

#include<bits/stdc++.h>
using namespace std;
#define N 50005
#define L (k<<1)
#define R (L+1)
#define mid (l+r>>1)
int n,m,p,x,y,f[N<<3],ls[N<<3],rs[N<<3],laz[N<<3];
void up(int k,int l,int r){
    if (laz[k]>=0)f[k]=ls[k]=rs[k]=laz[k]*(r-l+1);
    else{
        f[k]=max(max(f[L],f[R]),rs[L]+ls[R]);
        ls[k]=ls[L]+(ls[L]==(mid-l+1))*ls[R];
        rs[k]=rs[R]+(rs[R]==(r-mid))*rs[L];
    }
}
void down(int k,int l,int r){
    if (laz[k]<0)return;
    laz[L]=laz[R]=laz[k];
    f[L]=ls[L]=rs[L]=laz[L]*(mid-l+1);
    f[R]=ls[R]=rs[R]=laz[R]*(r-mid);
    laz[k]=-1;
}
void update(int k,int l,int r,int x,int y,int p){
    if ((l>y)||(x>r))return;
    if ((x<=l)&&(r<=y)){
        f[k]=ls[k]=rs[k]=p*(r-l+1);
        laz[k]=p;
        return;
    }
    down(k,l,r);
    update(L,l,mid,x,y,p);
    update(R,mid+1,r,x,y,p);
    up(k,l,r);
}
int query(int k,int l,int r,int x){
    down(k,l,r);
    if (f[L]>=x)return query(L,l,mid,x);
    if (rs[L]+ls[R]>=x)return mid-rs[L]+1;
    return query(R,mid+1,r,x);
}
int main(){
    scanf("%d%d",&n,&m);
    memset(laz,-1,sizeof(laz));
    update(1,1,n,1,n,1);
    for(int i=1;i<=m;i++){
        scanf("%d",&p);
        if (p==1){
            scanf("%d",&x);
            if (f[1]<x)y=0;
            else y=query(1,1,n,x);
            printf("%d
",y);
            if (y)update(1,1,n,y,y+x-1,0);
        }
        else{
            scanf("%d%d",&x,&y);
            update(1,1,n,x,x+y-1,1);
        }
    }
}