bzoj4825 [Hnoi2017]单旋（splay）

最新推荐文章于 2019-03-25 16:45:00 发布

Icefox_zhx

最新推荐文章于 2019-03-25 16:45:00 发布

阅读量381

点赞数

分类专栏： bzoj -------数据结构------- 平衡树线段树 STL

本文链接：https://blog.csdn.net/Icefox_zhx/article/details/79487053

版权

bzoj 同时被 3 个专栏收录

764 篇文章 1 订阅

订阅专栏

-------数据结构-------

251 篇文章 0 订阅

订阅专栏

线段树

123 篇文章 0 订阅

订阅专栏

这题真的是太神了orz
首先我们观察一下spaly单旋最值的性质。以把min单旋到根为例。

我们发现这一路上一定是不断的右旋，每次使得min深度-1，它的右子树深度不变，他的父亲及他的父亲的右子树深度+1。最终形态就是min成为了根，原来的根接在了min的右子树上，min原来的右子树接在了fa[min]的左子树上。除了min的右子树深度未变以外，其余节点深度均+1，min的深度变为1.单旋最大值同理。

我们发现每次操作以后其实对树的形态改变很小，对深度的影响就是个区间加法和单点修改。问题在于如何维护树的形态即如何快速的知道x的子树是哪些点。

还是对于mn来讨论，我们发现mn的右子树一定在权值大小上是一段连续的区间，即都处于区间 $[mn,v[fa[mn])$ .我们可以就按照权值大小去建这棵splay，每个点维护实际深度和区间最小深度。有区间加法和单点修改。如何找到mn的右子树呢？就是这棵splay上左起深度>=dep[mn]的连续区间。

还有一个问题，如何插入呢？插入值x时，肯定是插在x的前驱的右子树上或者x的后继的左子树上，而且不难发现这个位置是唯一的，x一定会插在前驱后继中较深的一个的子树上，因此我们可以直接得到这个插入的数的深度。

复杂度 $O(nlogn)$

注意该下放的时候一定要下放。。。

upd: 搞了搞线段树做法，先离散化，每个节点维护深度。其余的基本类似，用set来维护最小值最大值以及查找前驱后继即可。坑爹的add为什么会有x>y的情况呢？蒟蒻我怎么觉得不会有呢？绝望.jpg

splay

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define N 100010
#define ll long long
#define inf 0x3f3f3f3f
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int m,n=0,fa[N],c[N][2],sz[N],v[N],add[N],dep[N],mn[N],rt=0;
inline void update(int p){
    int l=c[p][0],r=c[p][1];
    sz[p]=sz[l]+sz[r]+1;
    mn[p]=min(min(mn[l],mn[r]),dep[p]);
}
inline void pushdown(int p){
    if(!add[p]) return;int l=c[p][0],r=c[p][1];
    if(l) add[l]+=add[p],mn[l]+=add[p],dep[l]+=add[p];
    if(r) add[r]+=add[p],mn[r]+=add[p],dep[r]+=add[p];add[p]=0;
}
inline void rotate(int x,int &k){
    int y=fa[x],z=fa[y],l=x==c[y][1],r=l^1;
    if(y==k) k=x;
    else c[z][y==c[z][1]]=x;
    fa[c[x][r]]=y;fa[y]=x;fa[x]=z;
    c[y][l]=c[x][r];c[x][r]=y;update(y);update(x);
}
inline void splay(int x,int &k){
    while(x!=k){
        int y=fa[x],z=fa[y];
        if(y!=k){
            if(x==c[y][1]^y==c[z][1]) rotate(x,k);
            else rotate(y,k);
        }rotate(x,k);
    }
}
inline void ins(int &p,int Fa){
    if(!p){p=n;fa[p]=Fa;splay(p,rt);return;}
    if(v[n]<v[p]) ins(c[p][0],p);
    else ins(c[p][1],p);
}
inline int find(int p,int k){
    pushdown(p);
    if(sz[c[p][0]]+1==k) return p;
    if(k<=sz[c[p][0]]) return find(c[p][0],k);
    return find(c[p][1],k-sz[c[p][0]]-1);
}
inline int getpre(int x){
    int p=rt,pre=0;
    while(p){
        pushdown(p);
        if(v[p]<x) pre=p,p=c[p][1];
        else p=c[p][0];
    }return pre;
}
inline int getsucc(int x){
    int p=rt,succ=0;
    while(p){
        pushdown(p);
        if(v[p]>x) succ=p,p=c[p][0];
        else p=c[p][1];
    }return succ;
}
inline int split(int xx,int yy){
    int x=find(rt,xx),y=find(rt,yy);
    splay(x,rt);splay(y,c[x][1]);return c[y][0];
}
inline void doadd(int l,int r,int val){
    int x=split(l-1,r+1),y=fa[x];
    add[x]+=val;mn[x]+=val;dep[x]+=val;update(y);update(fa[y]);
}
inline int qleft(int p,int x){//查左起深度>=x的点的个数
    if(!p) return 0;
    pushdown(p);
    if(mn[c[p][0]]>=x&&dep[p]>=x) return sz[c[p][0]]+1+qleft(c[p][1],x);
    return qleft(c[p][0],x);
}
inline int qright(int p,int x){//查右起深度>=x的点的个数
    if(!p) return 0;
    pushdown(p);
    if(mn[c[p][1]]>=x&&dep[p]>=x) return sz[c[p][1]]+1+qright(c[p][0],x);
    return qright(c[p][1],x);
}
inline void change(int p,int x){//把权值为x的点深度改为1.
    if(v[p]==x){dep[p]=1;update(p);return;}
    pushdown(p);
    if(x<v[p]) change(c[p][0],x);
    else change(c[p][1],x);update(p);
}
inline void del(int xx){
    int x=find(rt,xx-1),y=find(rt,xx+1);
    splay(x,rt);splay(y,c[x][1]);c[y][0]=fa[c[y][0]]=0;
    update(y);update(x);
}
int main(){
//  freopen("a.in","r",stdin);
    m=read();n=1;v[1]=-inf;dep[1]=mn[1]=inf;sz[1]=1;ins(rt,0);
    n=2;v[2]=inf;dep[2]=mn[2]=inf;sz[2]=1;ins(rt,0);mn[0]=inf;
    while(m--){
        int op=read();
        if(op==1){
            int x=read(),pre=getpre(x),succ=getsucc(x);++n;v[n]=x;
            mn[n]=dep[n]=max(pre>2?dep[pre]:0,succ>2?dep[succ]:0)+1;
            sz[n]=1;printf("%d\n",dep[n]);ins(rt,0);continue;
        }if((op&1)==0){
            int x=find(rt,2),y=min(qleft(rt,dep[x])-1,sz[rt]-2);
            printf("%d\n",dep[x]);doadd(2,sz[rt]-1,1);
            doadd(2,y+1,-1);change(rt,v[x]);
        }if(op&1){
            int x=find(rt,sz[rt]-1),y=min(qright(rt,dep[x])-1,sz[rt]-2);
            printf("%d\n",dep[x]);doadd(2,sz[rt]-1,1);
            doadd(sz[rt]-1-y+1,sz[rt]-1,-1);change(rt,v[x]);
        }if(op==4){del(2);doadd(2,sz[rt]-1,-1);}
        if(op==5){del(sz[rt]-1);doadd(2,sz[rt]-1,-1);}
    }return 0;
}

线段树+set

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <set>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 100010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(S==T) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n=0,m,a[N],fa[N],c[N][2],rt;
struct quer{
    int op,x;
}qq[N];
struct node{
    int dep;
}tr[N<<2];
set<int>st;
inline void pushdown(int p){
    if(!tr[p].dep) return;
    tr[p<<1].dep+=tr[p].dep;tr[p<<1|1].dep+=tr[p].dep;tr[p].dep=0;
}
inline int ask(int p,int l,int r,int x){
    if(l==r) return tr[p].dep;
    int mid=l+r>>1;pushdown(p);
    if(x<=mid) return ask(p<<1,l,mid,x);
    return ask(p<<1|1,mid+1,r,x);
}
inline void change(int p,int l,int r,int x,int val){
    if(l==r){tr[p].dep=val;return;}
    int mid=l+r>>1;pushdown(p);
    if(x<=mid) change(p<<1,l,mid,x,val);
    else change(p<<1|1,mid+1,r,x,val);
}
inline void add(int p,int l,int r,int x,int y,int val){
    if(x>y) return;
    if(x<=l&&r<=y){tr[p].dep+=val;return;}
    int mid=l+r>>1;pushdown(p);
    if(x<=mid) add(p<<1,l,mid,x,y,val);
    if(y>mid) add(p<<1|1,mid+1,r,x,y,val);
}
inline void ins(int x){
    if(st.empty()){st.insert(x);change(1,1,n,x,1);rt=x;puts("1");return;}
    set<int>::iterator it=st.lower_bound(x);
    if(it==st.begin()){
        int y=*it,depx=ask(1,1,n,y)+1;printf("%d\n",depx);
        change(1,1,n,x,depx);fa[x]=y;c[y][0]=x;st.insert(x);return;
    }if(it==st.end()){
        int y=*(st.rbegin()),depx=ask(1,1,n,y)+1;printf("%d\n",depx);
        change(1,1,n,x,depx);fa[x]=y;c[y][1]=x;st.insert(x);return;
    }int succ=*it,pre=*(--it);st.insert(x);
    int dpre=ask(1,1,n,pre),dsucc=ask(1,1,n,succ),depx;
    if(dpre>dsucc) change(1,1,n,x,dpre+1),fa[x]=pre,c[pre][1]=x,depx=dpre+1;
    else change(1,1,n,x,dsucc+1),fa[x]=succ,c[succ][0]=x,depx=dsucc+1;
    printf("%d\n",depx);
}
int main(){
//  freopen("a.in","r",stdin);
    m=read();
    for(int i=1;i<=m;++i){
        qq[i].op=read();if(qq[i].op==1) qq[i].x=a[++n]=read();
    }sort(a+1,a+n+1);
    for(int i=1;i<=m;++i){
        if(qq[i].op==1){
            int x=lower_bound(a+1,a+n+1,qq[i].x)-a;ins(x);continue;
        }if(qq[i].op==2){
            int x=*(st.begin()),y=fa[x];
            if(!y){puts("1");continue;}printf("%d\n",ask(1,1,n,x));
            add(1,1,n,y,n,1);change(1,1,n,x,1);
            c[y][0]=c[x][1];fa[c[x][1]]=y;c[x][1]=rt;fa[rt]=x;fa[x]=0;rt=x;
        }if(qq[i].op==3){
            int x=*(st.rbegin()),y=fa[x];
            if(!y){puts("1");continue;}printf("%d\n",ask(1,1,n,x));
            add(1,1,n,1,y,1);change(1,1,n,x,1);
            c[y][1]=c[x][0];fa[c[x][0]]=y;c[x][0]=rt;fa[rt]=x;fa[x]=0;rt=x;
        }if(qq[i].op==4){
            int x=*(st.begin()),y=fa[x];st.erase(x);
            if(!y){puts("1");add(1,1,n,1,n,-1);rt=c[x][1];fa[rt]=0;continue;}
            printf("%d\n",ask(1,1,n,x));
            add(1,1,n,1,y-1,-1);c[y][0]=c[x][1];fa[c[x][1]]=y;
        }if(qq[i].op==5){
            int x=*(st.rbegin()),y=fa[x];st.erase(x);
            if(!y){puts("1");add(1,1,n,1,n,-1);rt=c[x][0];fa[rt]=0;continue;}
            printf("%d\n",ask(1,1,n,x));
            add(1,1,n,y+1,n,-1);c[y][1]=c[x][0];fa[c[x][0]]=y;
        }
    }return 0;
}