浅谈平衡树splay

  • 首先splay和treap不一样,treap通过随机数来调整树的形态。但splay不一样,再每插入或操作一次后,你都会把他旋转到根,再旋转过程中树的形态会不断改变,这样来达到均摊效果 常数据大

  • 来看看具体实现吧
    首先定义数组,\(size\) 子树大小(包括自己),\(fa\) 节点的父亲,\(key\) 该节点的权值,\(cnt\) 该节点权值出现次数,$ch $表示儿子 0表左二子,1表右儿子

首先看几个简单函数

inline void update(int x)
{
	size[x]=cnt[x]+size[ch[x][0]]+size[ch[x][1]];
}

更新子树大小

inline int get(int x){return x==ch[fa[x]][1];}

返回该节点是left儿子还是right儿子

inline void clear(int x){ch[x][0]=ch[x][1]=fa[x]=size[x]=cnt[x]=key[x]=0;}

删除该节点,清空所有信息

接下来是splay的精髓所在

inline void rotate(int x,int &k)
{
    static int old,oldfa,o;
    old=fa[x];oldfa=fa[old];o=get(x);
    if(old==k)k=x;
    else ch[oldfa][get(old)]=x;
    fa[x]=oldfa;
    ch[old][o]=ch[x][o^1];fa[ch[x][o^1]]=old;
    ch[x][o^1]=old;fa[old]=x;
    update(x),update(old);
}
inline void splay(int x,int &k)
{
    while(x!=k)
    {
        if(fa[x]!=k)rotate(get(x)^get(fa[x])?x:fa[x],k);
        rotate(x,k);
    }
}

rotate,splay,是splay核心操作,显然splay是依赖于rotate的,让我们看一下rotate是如何实现的吧
image
(手绘图)
我们考虑从图上左往右的过程,我们要将y旋上去,因为y本是x的右儿子,所以x放到y的左儿子,将y的原本左儿子设为x的右儿子,这是左旋,还有对称操作右旋,但我们不必要打两个函数,用 ^可以实现左右儿子的转换,用get操作实现,具体实现参考代码,打代码时最好画个图参照一下。

splay,这个操作完全依靠rotate,目的就是把你要的节点旋转到k(一般是root),k要传地址,要修改。在while循环里加了个小小的优化,但x和他的fa在同一侧时可以旋fa,以此来改变树的形态,(不怕被卡可以不写)

inline void insert(int x)
{
    if(!root){root=++sz;size[sz]=cnt[sz]=1;key[sz]=x;return;}
    int now=root,o;
    while(1)
    {
        if(x==key[now])
        {
            ++cnt[now];
            splay(now,root);
            update(now);
			return;
        }
        o=x>key[now]?1:0;
        if(!ch[now][o])
        {
            ch[now][o]=++sz;
            size[sz]=cnt[sz]=1;
            key[sz]=x;fa[sz]=now;
            splay(sz,root);
            return;
        }
        else now=ch[now][o];
    }
}

insert,插入一个数,当没有数时就直接把这个数设为根,else 因为树满足二叉排序树的性质,所以比当前节点的key小就往左走,否则往右走,直到找到一个空节点,更新信息,由于这个点以上所有的点\(size\)都要加一,不好update,所以把这给点旋转到根,将这个点update就行了

inline int find_pos(int x)
{
    int now=root;
    while(1)
    {
        if(x==key[now]){return now;}
        if(x<key[now])now=ch[now][0];
        else now=ch[now][1];
    }
}

找到该值在树中的节点编号

inline int pre()
{
        int now=ch[root][0];
	while(ch[now][1])now=ch[now][1];
	return now;
}
inline int nex()
{
        int now=ch[root][1];
	while(ch[now][0])now=ch[now][0];
	return now;
}

求前驱,后继,前驱从根的左儿子开始一直往右跑,后继从根的右儿子开始一直往左跑即可

void del(int x)
{
    splay(find_pos(x),root);
    if(cnt[root]>1){--cnt[root];return;}
    if(!ch[root][0]&&!ch[root][1]){clear(root);root=0;return;}
    if(ch[root][0]&&ch[root][1])
    {
        int oldroot=root;
        splay(pre(),root);
        fa[ch[oldroot][1]]=root;
        ch[root][1]=ch[oldroot][1];
        clear(oldroot);
        update(root);
    }
    else
    {
        int o=ch[root][1]>0;
        root=ch[root][o];
        clear(fa[root]);
        fa[root]=0;
    }
}

删除操作,有点麻烦,先找到x的位置

  • 如果x有多个就\(cnt\)减一
  • 如果一个儿子都没有就直接删掉,root设为0
  • 如果 只有一个儿子就把儿子设为根,删去这个点
  • 剩下两个儿子情况,找到根的前驱,把前驱旋到根,这是root只有左儿子,再把原来根的右儿子到root上,这样原来的root就脱离了树,再删掉即可。
inline int find_order_of_key(int x)
{
    int res=0,now=root;
    while(1)
    {
        if(x<key[now])now=ch[now][0];
        else
        {
            res+=size[ch[now][0]];
            if(x==key[now]){splay(now,root);return res+1;}
            res+=cnt[now];
            now=ch[now][1];
        }
    }
}
inline int find_by_order(int x)
{
    int now=root;
    while(1)
    {
        if(x<=size[ch[now][0]])now=ch[now][0];
        else
        {
            int temp=size[ch[now][0]]+cnt[now];
            if(x<=temp)return key[now];
            else{x-=temp;now=ch[now][1];}
        }
    }
}

找x的排名,与找排名为x的数,其实大同小异,用二叉搜索树的性质即可,只是记得答案不一样罢了

inline void rever(int x)
{
    swap(ch[x][0],ch[x][1]);
    rev[ch[x][0]]^=1;rev[ch[x][1]]^=1;
    rev[x]=0;
}
inline void rever(int l,int r)
{
	l=find(l-1);r=find(r+1);
        splay(l,root);splay(r,ch[l][1]);
        rev[ch[r][0]]^=1;
}

找到区间左边一个和区间的右边一个点在树中位置,把左边的点旋转到根,再把右边的点旋到root的右儿子,这时这段区间一定是ch[r][0]的子树(想一想,为什么)(根据二叉搜索树的性质),把这个点打上标记即可;当遇到有翻转标记的点时,交换其左右子树,并下传标记即可,注意,翻转操作只有可能在维护无序数列时使用,在有序数列中不需要也不能翻转,不然就无法满足排序二叉树的性质。
翻转操作在找节点的编号时才执行,详见下面代码

ok,splay的基本操作就是这些了

下面是完整代码
洛谷P3369 treap模板
题目描述
您需要写一种数据结构(可参考题目标题),来维护一些数,其中需要提供以下操作:

插入x数

删除x数(若有多个相同的数,因只删除一个)

查询x数的排名(排名定义为比当前数小的数的个数+1。若有多个相同的数,因输出最小的排名)

查询排名为x的数

求x的前驱(前驱定义为小于x,且最大的数)
求x的后继(后继定义为大于x,且最小的数)
#include<bits/stdc++.h>
using namespace std;
typedef int sign;
typedef long long ll;
#define For(i,a,b) for(register sign i=(sign)a;i<=(sign)b;++i)
#define Fordown(i,a,b) for(register sign i=(sign)a;i>=(sign)b;--i)
const int N=1e5+5;
void cmax(sign &a,sign b){if(a<b)a=b;}
void cmin(sign &a,sign b){if(a>b)a=b;}
template<typename T>T read()
{
    T ans=0,f=1;
    char ch=getchar();
    while(!isdigit(ch)&&ch!='-')ch=getchar();
    if(ch=='-')f=-1,ch=getchar();
    while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch-'0'),ch=getchar();
    return ans*f;
}
void file()
{
    #ifndef ONLINE_JUDGE
        freopen("splay.in","r",stdin);
        freopen("splay.out","w",stdout);
    #endif
}
int fa[N],size[N],key[N],cnt[N],ch[N][2],sz,root;
inline void update(int x){size[x]=cnt[x]+size[ch[x][0]]+size[ch[x][1]];}
inline int get(int x){return x==ch[fa[x]][1];}
inline void clear(int x){ch[x][0]=ch[x][1]=fa[x]=size[x]=cnt[x]=key[x]=0;}
inline void rotate(int x,int &k)
{
    static int old,oldfa,o;
    old=fa[x];oldfa=fa[old];o=get(x);
    if(old==k)k=x;
    else ch[oldfa][get(old)]=x;
    fa[x]=oldfa;
    ch[old][o]=ch[x][o^1];fa[ch[x][o^1]]=old;
    ch[x][o^1]=old;fa[old]=x;
    update(x),update(old);
}
inline void splay(int x,int &k)
{
    while(x!=k)
    {
        if(fa[x]!=k)rotate(get(x)^get(fa[x])?x:fa[x],k);
        rotate(x,k);
    }
}
inline void insert(int x)
{
    //puts("");
    if(!root){root=++sz;size[sz]=cnt[sz]=1;key[sz]=x;return;}
    int now=root,o;
    while(1)
    {
        if(x==key[now])
        {
            ++cnt[now];
            splay(now,root);
            update(now);
            return;
        }
        o=x>key[now]?1:0;
        if(!ch[now][o])
        {
            ch[now][o]=++sz;
            size[sz]=cnt[sz]=1;
            key[sz]=x;fa[sz]=now;
            splay(sz,root);
            return;
        }
        else now=ch[now][o];
        //printf("%d %d %d %d\n",now,fa[now],ch[now][0],ch[now][1]);
    }
}
inline int find_pos(int x)
{
    int now=root;
    while(1)
    {
        if(x==key[now]){return now;}
        if(x<key[now])now=ch[now][0];
        else now=ch[now][1];
    }
}
inline int pre()
{
    int now=ch[root][0];
    while(ch[now][1])now=ch[now][1];
    return now;
}
inline int nex()
{
    int now=ch[root][1];
    while(ch[now][0])now=ch[now][0];
    return now;
}
void del(int x)
{
    splay(find_pos(x),root);
    if(cnt[root]>1){--cnt[root];return;}
    if(!ch[root][0]&&!ch[root][1]){clear(root);root=0;return;}
    if(ch[root][0]&&ch[root][1])
    {
        int oldroot=root;
        splay(pre(),root);
        fa[ch[oldroot][1]]=root;
        ch[root][1]=ch[oldroot][1];
        clear(oldroot);
        update(root);
    }
    else
    {
        int o=ch[root][1]>0;
        root=ch[root][o];
        clear(fa[root]);
        fa[root]=0;
    }
}
inline int find_order_of_key(int x)
{
    int res=0,now=root;
    while(1)
    {
        if(x<key[now])now=ch[now][0];
        else
        {
            res+=size[ch[now][0]];
            if(x==key[now]){splay(now,root);return res+1;}
            res+=cnt[now];
            now=ch[now][1];
        }
    }
}
inline int find_by_order(int x)
{
    int now=root;
    while(1)
    {
        if(x<=size[ch[now][0]])now=ch[now][0];
        else
        {
            int temp=size[ch[now][0]]+cnt[now];
            if(x<=temp)return key[now];
            else{x-=temp;now=ch[now][1];}
        }
    }
}
void input()
{
    int T=read<int>();
    int opt,x;
    while(T--)
    {
        opt=read<int>();x=read<int>();
        if(opt==1)insert(x);
        else if(opt==2)del(x);
        else if(opt==3)printf("%d\n",find_order_of_key(x));
        else if(opt==4)printf("%d\n",find_by_order(x));
        else if(opt==5)
        {
            insert(x);
            printf("%d\n",key[pre()]);
            del(x);
        }
        else if(opt==6)
        {
            insert(x);
            printf("%d\n",key[nex()]);
            del(x);
        }
    }
}
int main()
{
    file();
    input();
    return 0;
}

洛谷P3391 splay模板
题目描述
您需要写一种数据结构(可参考题目标题),来维护一个有序数列,其中需要提供以下操作:翻转一个区间,输出一行n个数字,表示原始序列经过m次变换后的结果

#include<bits/stdc++.h>
using namespace std;
typedef int sign;
typedef long long ll;
#define For(i,a,b) for(register sign i=(sign)a;i<=(sign)b;++i)
#define Fordown(i,a,b) for(register sign i=(sign)a;i>=(sign)b;--i)
const int N=1e5+5;
bool cmax(sign &a,sign b){return (a<b)?a=b,1:0;}
bool cmin(sign &a,sign b){return (a>b)?a=b,1:0;}
template<typename T>T read()
{
    T ans=0,f=1;
    char ch=getchar();
    while(!isdigit(ch)&&ch!='-')ch=getchar();
    if(ch=='-')f=-1,ch=getchar();
    while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch-'0'),ch=getchar();
    return ans*f;
}
void file()
{
    #ifndef ONLINE_JUDGE
        freopen("splay.in","r",stdin);
        freopen("splay.out","w",stdout);
    #endif
}
int ch[N][2],fa[N],size[N],rev[N],root,sz;
inline int get(int x){return x==ch[fa[x]][1];}
inline void update(int x){size[x]=1+size[ch[x][0]]+size[ch[x][1]];}
inline void rotate(int x,int &k)
{
    int old=fa[x],oldfa=fa[old],o=get(x);
    if(k==old)k=x;
    else ch[oldfa][ch[oldfa][1]==old]=x;
    fa[x]=oldfa;fa[old]=x;fa[ch[x][o^1]]=old;
    ch[old][o]=ch[x][o^1];ch[x][o^1]=old;
    update(x),update(old);
}
inline void splay(int x,int &k)
{
    while(x!=k)
    {
        if(fa[x]!=k)rotate(get(x)^get(fa[x])?x:fa[x],k);
        //printf("%d %d\n",x,k);
        rotate(x,k);
    }
}
#define mid ((l+r)>>1)
inline void build(int l,int r,int pre)
{
    if(l>r)return;
    ch[pre][mid>=pre]=mid;	
    fa[mid]=pre;size[mid]=1;
    if(l==r)return;
    build(l,mid-1,mid);build(mid+1,r,mid);
    update(mid);
}
#undef mid  
int n,m;
void input(){n=read<int>();m=read<int>();}
inline void rever(int x)
{
    swap(ch[x][0],ch[x][1]);
    rev[ch[x][0]]^=1;rev[ch[x][1]]^=1;
    rev[x]=0;
}
int find(int x)
{
    int now=root;
    while(1)
    {
        if(rev[now])rever(now);
        if(size[ch[now][0]]>=x)now=ch[now][0];
        else 
        {
            if(size[ch[now][0]]==x-1)return now;
            x=x-size[ch[now][0]]-1;
            now=ch[now][1];
        }
    }
}
void work()
{
    int l,r;
    root=(n+3)>>1;
    build(1,n+2,root);
    fa[root]=0;
    while(m--)
    {
        l=read<int>();r=read<int>();
        l=find(l);r=find(r+2);
        splay(l,root);splay(r,ch[l][1]);
        rev[ch[r][0]]^=1;
    }
}
void out(int x)
{
    if(rev[x])rever(x);
    if(ch[x][0])out(ch[x][0]);
    if(x>1&&x<n+2)printf("%d ",x-1);
    if(ch[x][1])out(ch[x][1]);
}
int main()
{
    file();
    input();
    work();
    out(root);
    return 0;
}

另外推荐一篇写得好的博客

posted @ 2017-11-26 08:39  dyx_diversion  阅读(1193)  评论(0编辑  收藏  举报