bzoj 1593: [Usaco2008 Feb]Hotel 旅馆【线段树】

参考:https://blog.csdn.net/u010336344/article/details/53034372
神一样的线段树
线段树上维护:ll从左开始最长空段;rr从右开始最长空段;len区间中最长空段;tg:-1不全是空的,0区间内全是空的,1区间内全是满的;lz下传标记:-1没标记,1下传满的,0下传空的
修改的时候,修改到一整个区间,就把lz和tg改了,ll、rr、len全改成r-l+1或者0
然后下传标记的时候同上
向上合并的时候,先把左右区间的ll、rr穿给当前区间,然后根据左右区间的tg是否为0(全空)来扩大当前区间的ll、rr;tg根据左右区间tg更改;len从左区间len、右区间len和左区间rl+右区间ll中取max
查询的时候优先查左边即可,像二分一样在树上跳

#include<iostream>
#include<cstdio>
using namespace std;
const int N=100005;
int n,m;
struct xds
{
    int l,r,ll,rl,tg,len,lz;
}t[N<<1];
int read()
{
    int r=0,f=1;
    char p=getchar();
    while(p>'9'||p<'0')
    {
        if(p=='-')
            f=-1;
        p=getchar();
    }
    while(p>='0'&&p<='9')
    {
        r=r*10+p-48;
        p=getchar();
    }
    return r*f;
}
void pd(int ro)
{
    if(t[ro].lz!=-1)
    {
        if(t[ro].lz==1)
        {
            t[ro<<1].rl=t[ro<<1].ll=t[ro<<1].len=0;
            t[ro<<1|1].rl=t[ro<<1|1].ll=t[ro<<1|1].len=0;
        }
        else
        {
            t[ro<<1].rl=t[ro<<1].ll=t[ro<<1].len=t[ro<<1].r-t[ro<<1].l+1;
            t[ro<<1|1].rl=t[ro<<1|1].ll=t[ro<<1|1].len=t[ro<<1|1].r-t[ro<<1|1].l+1;
        }
        t[ro<<1].tg=t[ro<<1|1].tg=t[ro].tg;
        t[ro<<1].lz=t[ro<<1|1].lz=t[ro].lz;
        t[ro].lz=-1;
    }
}
void ud(int ro)
{
    if(t[ro<<1].tg==t[ro<<1|1].tg)
        t[ro].tg=t[ro<<1].tg;
    else
        t[ro].tg=-1;
    t[ro].ll=t[ro<<1].ll;
    t[ro].rl=t[ro<<1|1].rl;
    if(!t[ro<<1].tg)
        t[ro].ll+=t[ro<<1|1].ll;
    if(!t[ro<<1|1].tg)
        t[ro].rl+=t[ro<<1].rl;
    t[ro].len=max(max(t[ro<<1].len,t[ro<<1|1].len),t[ro<<1].rl+t[ro<<1|1].ll);
}
void build(int ro,int l,int r)
{
    t[ro].l=l,t[ro].r=r,t[ro].len=t[ro].ll=t[ro].rl=r-l+1,t[ro].lz=-1;
    if(l==r)
        return;
    int mid=(l+r)>>1;
    build(ro<<1,l,mid);
    build(ro<<1|1,mid+1,r);
}
void update(int ro,int l,int r,int k)
{
    if(t[ro].l==l&&t[ro].r==r)
    {
        if(k)
            t[ro].rl=t[ro].ll=t[ro].len=0;
        else
            t[ro].rl=t[ro].ll=t[ro].len=r-l+1;
        t[ro].tg=t[ro].lz=k;
        return;
    }
    pd(ro);
    int mid=(t[ro].l+t[ro].r)>>1;
    if(r<=mid)
        update(ro<<1,l,r,k);
    else if(l>mid)
        update(ro<<1|1,l,r,k);
    else
        update(ro<<1,l,mid,k),update(ro<<1|1,mid+1,r,k);
    ud(ro);
}
int ques(int ro,int k)
{
    int re=-1;
    while(t[ro].l&&t[ro].len>=k)
    {
        if(t[ro].ll>=k)
        {
            re=t[ro].l;
            break;
        }
        if(t[ro<<1].len>=k)
            ro<<=1;
        else
        {
            if(t[ro<<1].rl!=0&&t[ro<<1].rl+t[ro<<1|1].ll>=k)
            {
                re=t[ro<<1].r-t[ro<<1].rl+1;
                break;
            }
            else
                ro=ro<<1|1;
        }
    }
    return re;
}
int main()
{
    n=read(),m=read();
    build(1,1,n);
    while(m--)
    {
        int o=read();
        if(o==1)
        {
            int x=read(),now=ques(1,x);
            if(now!=-1)
            {
                printf("%d\n",now);
                update(1,now,now+x-1,1);
            }
            else
                puts("0");
        }
        else
        {
            int x=read(),y=read();
            update(1,x,x+y-1,0);
        }
    }
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/lokiii/p/8992537.html
今日推荐