[POJ3667]Hotel（线段树）-优快云博客

=== ===

题解

这道题的意思是动态维护连续为0的个数然后返回最靠左的长度大于给定的x的一段连续的0区间。记得当时ATP它们老师考这个题的时候ATP一时脑抽差点写出平衡树来23333实际上这是一个线段树的经典问题啦。

只要在线段树的节点里面维护三个量——最靠左的连续区间，最靠右的连续区间和总的连续区间就可以了。反正就是在update的时候麻烦的讨论一下判断一下左右子树的继承关系。。在查找的时候，find过程返回找到的区间的左端点。仍然先判断左子树的tot是否大于x，是的话就进入左子树查找，否则判断这个区间是否跨了两个半区间，也就是左子树的right加上右子树的left；再否则就进入右子树查询。

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,m,ans,L,R;
struct segtree{
    int tot,lt,rt,dlt;
    segtree(){tot=lt=rt=0;dlt=-1;}
}t[2000010];
void update(int i,int l,int r){
    int mid=(l+r)>>1;
    t[i].lt=t[i<<1].lt;
    t[i].rt=t[(i<<1)+1].rt;
    if (t[i<<1].lt==mid-l+1) t[i].lt+=t[(i<<1)+1].lt;//注意判断是否扩展
    if (t[(i<<1)+1].rt==r-mid) t[i].rt+=t[i<<1].rt;
    t[i].tot=max(max(t[i<<1].tot,t[(i<<1)+1].tot),t[i<<1].rt+t[(i<<1)+1].lt);
}
void pushdown(int i,int l,int r){
    if (t[i].dlt!=-1){
        int mid=(l+r)>>1,lc=i<<1,rc=(i<<1)+1;
        t[lc].dlt=t[i].dlt;
        t[lc].tot=t[lc].lt=t[lc].rt=t[i].dlt*(mid-l+1);
        t[rc].dlt=t[i].dlt;
        t[rc].tot=t[rc].lt=t[rc].rt=t[i].dlt*(r-mid);
        t[i].dlt=-1;
    }
}
void build(int i,int l,int r){
    if (l==r){
        t[i].tot=t[i].lt=t[i].rt=1;
        t[i].dlt=-1;
        return;
    }
    int mid=(l+r)>>1;
    build(i<<1,l,mid);
    build((i<<1)+1,mid+1,r);
    update(i,l,r);
}
void change(int i,int l,int r,int left,int right,int v){
    if ((left<=l)&&(right>=r)){
        t[i].tot=t[i].lt=t[i].rt=v*(r-l+1);
        t[i].dlt=v;
        return;
    }
    int mid=(l+r)>>1;
    pushdown(i,l,r);
    if (left<=mid)
      change(i<<1,l,mid,left,right,v);
    if (right>mid)
      change((i<<1)+1,mid+1,r,left,right,v);
    update(i,l,r);
}
int find(int i,int l,int r,int x){
    if (l==r) return l;
    int mid=(l+r)>>1;
    pushdown(i,l,r);
    if (t[i<<1].tot>=x)
      return find(i<<1,l,mid,x);
    else
      if (t[i<<1].rt+t[(i<<1)+1].lt>=x)
        return mid-t[i<<1].rt+1;
      else return find((i<<1)+1,mid+1,r,x);
}
int main()
{
    scanf("%d%d",&n,&m);
    build(1,1,n);
    for (int i=1;i<=m;i++){
        int o;
        scanf("%d",&o);
        if (o==1){
            int x;
            scanf("%d",&x);
            if (t[1].tot<x) printf("0\n");
            else{
                L=find(1,1,n,x);
                printf("%d\n",L);
                R=L+x-1;
                change(1,1,n,L,R,0);
            }
        }else{
            int x,y;
            scanf("%d%d",&x,&y);
            change(1,1,n,x,x+y-1,1);
        }
    }
    return 0;
}