Splay初学习

例题传送门

听YZ哥哥说Splay是一种很神奇的数据结构,所以学习了一下它的最基本操作。O(1)的Spaly。

伸展树(Splay Tree),也叫分裂树,是一种二叉排序树,它能在O(logn)内完成插入、查找和删除操作。它由丹尼尔·斯立特Daniel Sleator和罗伯特·恩卓·塔扬Robert Endre Tarjan在1985年发明的。
在伸展树上的一般操作都基于伸展操作:假设想要对一个二叉查找树执行一系列的查找操作,为了使整个查找时间更小,被查频率高的那些条目就应当经常处于靠近树根的位置。于是想到设计一个简单方法, 在每次查找之后对树进行重构,把被查找的条目搬移到离树根近一些的地方。伸展树应运而生。伸展树是一种自调整形式的二叉查找树,它会沿着从某个节点到树根之间的路径,通过一系列的旋转把这个节点搬移到树根去。
它的优势在于不需要记录用于平衡树的冗余信息。
Splay是一种自调整数据结构,它的核心是Splay操作。
Splay操作是Rotate的升级版,该函数将子节点旋转到根,来保持Splay的复杂度。
//我们这里讲的是双旋Splay,之所以称为双旋,是因为在判断父子关系之前,要旋一次,共旋两次。
 
Splay操作要分两种情况讨论。
①当前要Splay的点和它的父亲及其父亲的父亲(如果它有父亲的父亲的话)在同一直线上,先旋它的父亲。
②当前要Splay的点和它的父亲及其父亲的父亲(如果它有父亲的父亲的话)不在在同一直线上,旋该节点。
 
插入:在插入节点后,记录插入节点的位置,Splay到根。
删除:先找到要删的节点,记录下来,没有就return。删除要分几种情况讨论;
        ①没有左右节点,直接clear根。
      ②只有左节点或右节点,用左或右节点将根覆盖,clear原来的根。
      ③既有左节点又有右节点,将root的前驱Splay到根,然后将root的右节点(原来的root)的右节点连接到root的右节点上就好了。
其他操作跟Treap相似。
 
code:
#include <cstdio>
#include <cstring>
using namespace std;

int read()
{
    char c;while(c=getchar(),(c<'0'||c>'9')&&c!='-');
    int x=0,y=1;c=='-'?y=-1:x=c-'0';
    while(c=getchar(),c>='0'&&c<='9')x=x*10+c-'0';
    return x*y;
}

int N,dist;

struct Splay{
    int tr[100005][2],v[100005],tot[100005];
    int f[100005],fa[100005],root,cnt;
    Splay(){
        memset(tr,0,sizeof tr);
        memset(v,0,sizeof v);
        memset(tot,0,sizeof tot);
        memset(f,0,sizeof f);
        cnt=root=0;
    }//初始化
    
    void clear(int x){tr[x][0]=tr[x][1]=f[x]=fa[x]=tot[x]=v[x]=0;}//清除节点信息
    void up(int x){f[x]=f[tr[x][0]]+f[tr[x][1]]+tot[x];}//更新节点信息
    int get(int x){return tr[fa[x]][1]==x;}//which son of father
    
    void rotate(int &x)
    {
        int ol=fa[x],olol=fa[ol],to=get(x);
        tr[ol][to]=tr[x][to^1];fa[tr[x][to^1]]=ol;
        fa[ol]=x;tr[x][to^1]=ol;
        fa[x]=olol;
        if(olol)//如果该节点的父亲的父亲存在
            tr[olol][tr[olol][1]==ol]=x;
        up(ol);up(x);//这里一定要先更新ol,在更新x,想想为什么
    }
    
    void splay(int x)
    {
        for(int S;S=fa[x];rotate(x))//先旋x
            if(fa[S])//S不为根
                rotate((get(x)==get(S)?S:x));//判断是否三点一线
        root=x;
    }
    
    void insert(int &x,int val,int pos)
    {
        if(!x){//插入
            x=++cnt;
            f[x]=tot[x]=1,v[x]=val,fa[x]=pos;
            dist=x;//记录节点
            return ;
        }
        if(val==v[x]){dist=x,tot[x]++;return ;}//有一样的数
        insert(tr[x][val>v[x]],val,x);
        up(x);//更新
        return ;
    }
    
    int QueryX(int x,int val)//查询x的排名
    {
        if(!x)return 0;
        if(val==v[x])return f[tr[x][0]]+1;
        int to=val>v[x];
        return QueryX(tr[x][to],val)+(to?f[tr[x][0]]+tot[x]:0); 
    }
    int QueryK(int x,int kth)//查询排名为K的数
    {
        if(!x)return 0;
        if(kth<=f[tr[x][0]])return QueryK(tr[x][0],kth);
        if(kth>f[tr[x][0]]+tot[x])return QueryK(tr[x][1],kth-(f[tr[x][0]]+tot[x]));
        return v[x];
    }
    
    void pre(int x,int val)//前驱
    {
        if(!x)return ;
        if(val<=v[x])return pre(tr[x][0],val);
        else dist=x,pre(tr[x][1],val);
    }
    void bac(int x,int val)//后继
    {
        if(!x)return ;
        if(v[x]<=val)bac(tr[x][1],val);
        else dist=x,bac(tr[x][0],val);
    }
    
    int find(int x,int val)//在del之前先找节点,记录,并Splay
    {
        if(!x)return 0;
        if(v[x]==val){splay(x);return x;}//找到
        int to=val>v[x];
        return find(tr[x][to],val);
    }
    void del(int x)//删除
    {
        int kkk=find(root,x);
        if(!kkk)return ;//不存在
        if(tot[root]>1){
            tot[root]--,up(root);
            return ;
        }
        if(!tr[root][0]&&!tr[root][1]){
            clear(root),root=0;
            return ;
        }
        if(!(tr[root][0]*tr[root][1])){
            int rt=root;
            root=tr[root][0]+tr[root][1];
            fa[root]=0,clear(rt);//细节是要先记录root,再更新root,再将原来的root clear
            return ;
        }
        int ok=root;//原来的root
        pre(root,x);//求前驱
        splay(dist);//Splay到根
        fa[tr[ok][1]]=root;
        tr[root][1]=tr[ok][1];
        clear(ok);up(root);fa[root]=0;//clear,更新,细节是要把新根的fa清零
        return ;
    }
    
}W;

int main()
{
    N=read();
        while(N--){
            int o=read(),x;
                switch(o){
                    case 1:W.insert(W.root,read(),0);W.splay(dist);break;
                    case 2:W.del(read());break;
                    case 3:printf("%d\n",W.QueryX(W.root,read()));break;
                    case 4:printf("%d\n",W.QueryK(W.root,read()));break;
                    case 5:dist=0,W.pre(W.root,read());printf("%d\n",W.v[dist]);break;
                    case 6:dist=0,W.bac(W.root,read());printf("%d\n",W.v[dist]);break;
                }
        }
}

 

posted @ 2018-03-09 08:40  Cptraser  阅读(228)  评论(0编辑  收藏  举报