【洛谷P2894】Hotel 线段树+二分查询

题目大意：给定一个长度为 N 的序列，每个点有两种状态 1/0，表示占有和空闲，现支持 first-fit 查询是否有一段连续的长度为 X 的空闲子序列和区间赋值操作。

题解：get到了线段树新技能。。qwq
区间赋值操作和其他线段树一样，维护标记即可。
查询是否有一段连续的长度为 X 的空闲子序列只需要类似于维护区间连续最长字段即可，即：lmx,rmx,mx。first-fit 查询操作需要类似于在线段树上二分的操作，即：若左子树满足条件就走左子树，左右子树能拼接成满足需求的结果则返回，否则遍历右子树。

代码如下

#include <bits/stdc++.h>
using namespace std;
const int maxn=5e4+10;

int n,m,opt;
struct node{
    #define ls(o) t[o].lc
    #define rs(o) t[o].rc
    int lc,rc;
    int mx,lmx,rmx,tag; // 1 -> all zero    2 -> all one
}t[maxn<<1];
int tot,root;
inline void pushup(int o,int l,int r){
    int mid=l+r>>1;
    t[o].lmx=t[ls(o)].lmx==mid-l+1?t[ls(o)].lmx+t[rs(o)].lmx:t[ls(o)].lmx;
    t[o].rmx=t[rs(o)].rmx==r-mid?t[ls(o)].rmx+t[rs(o)].rmx:t[rs(o)].rmx;
    t[o].mx=max(t[ls(o)].rmx+t[rs(o)].lmx,max(t[ls(o)].mx,t[rs(o)].mx));
}
inline void pushdown(int o,int l,int r){
    if(!t[o].tag)return;
    int mid=l+r>>1;
    t[ls(o)].tag=t[rs(o)].tag=t[o].tag;
    t[ls(o)].lmx=t[ls(o)].rmx=t[ls(o)].mx=t[o].tag==1?mid-l+1:0;
    t[rs(o)].lmx=t[rs(o)].rmx=t[rs(o)].mx=t[o].tag==1?r-mid:0;
    t[o].tag=0;
}
int build(int l,int r){
    int o=++tot;
    if(l==r){t[o].lmx=t[o].rmx=t[o].mx=1;return o;}
    int mid=l+r>>1;
    ls(o)=build(l,mid),rs(o)=build(mid+1,r);
    pushup(o,l,r);
    return o;
}
void modify(int o,int l,int r,int x,int y,int flag){
    if(l==x&&r==y){
        t[o].lmx=t[o].rmx=t[o].mx=flag==1?r-l+1:0;
        t[o].tag=flag;
        return;
    }
    int mid=l+r>>1;
    pushdown(o,l,r);
    if(y<=mid)modify(ls(o),l,mid,x,y,flag);
    else if(x>mid)modify(rs(o),mid+1,r,x,y,flag);
    else modify(ls(o),l,mid,x,mid,flag),modify(rs(o),mid+1,r,mid+1,y,flag);
    pushup(o,l,r);
}
int query(int o,int l,int r,int len){
    if(l==r)return l;
    int mid=l+r>>1;
    pushdown(o,l,r);
    if(len<=t[ls(o)].mx)return query(ls(o),l,mid,len);
    else if(len<=t[ls(o)].rmx+t[rs(o)].lmx)return mid-t[ls(o)].rmx+1;
    else return query(rs(o),mid+1,r,len);
}

int main(){
    scanf("%d%d",&n,&m);
    root=build(1,n);
    while(m--){
        scanf("%d",&opt);
        if(opt==1){
            int x;
            scanf("%d",&x);
            if(t[root].mx<x)puts("0");
            else{
                int l=query(root,1,n,x);
                modify(root,1,n,l,l+x-1,2);
                printf("%d\n",l);
            }
        }else if(opt==2){
            int x,y;
            scanf("%d%d",&x,&y);
            modify(root,1,n,x,x+y-1,1);
        }
    }
    return 0;
}

转载于:https://www.cnblogs.com/wzj-xhjbk/p/10952095.html