首页 > 其他 > 详细

神仙数据结构——FHQ_Treap

时间:2018-11-07 23:42:28      阅读:181      评论:0      收藏:0      [点我收藏+]

$FHQ\_Treap$是平衡树的一种,它不仅支持几乎所有的平衡树的操作,而且实现特别简单,总共只有两个操作。这里来简单介绍一下。

基本操作

$FHQ\_Treap$和$Treap$一样是需要用随机值来维护树的形态的,但是$FHQ\_Treap$不需要旋转来调整形态,而且用$Split$和$Merge$来实现,也就是分离与合并,也就是这两种操作,完成了平衡树的所有操作。

分离(Split)

把一棵平衡树分离成两颗,从而方便插入和删除。分离有两种做法,一种是按权值分离,一种是按子树大小分离,根据情况选择。

权值分离版:

 

void split(int rt,int k,int &x,int &y)
{
    if( !rt ) x=y=0;
    else {
        if( val[rt]<=k ) x=rt, split(ch[rt][1],k,ch[rt][1],y);
        else y=rt, split(ch[rt][0],k,x,ch[rt][0]);
        pushup(rt);
    }
}

 

子树大小版:

 

void split(int rt,int k,int &x,int &y)
{
    if( !rt ) x=0, y=0;
    else {
        pushdown(rt);
        if( siz[ch[rt][0]]<k ) {
            x=rt, split(ch[rt][1],k-siz[ch[rt][0]]-1,ch[rt][1],y);
        } else {
            y=rt, split(ch[rt][0],k,x,ch[rt][0]);
        }
        pushup(rt);
    }
}

合并(merge)

合并操作与左偏树的合并类似,按照优先级合并,把优先级高的作为父节点。当然,合并的时候,函数中的两个参数$x,y$要按照顺序,$x$是权值较小的,$y$是权值较大的,因为合并的时候是按照随机值来确定优先级的。(当然在函数中交换也行,不过常数大些)

 

int merge(int x,int y)
{
    if( !x || !y ) return x+y;
    pushdown(x), pushdown(y);
    if( p[x]<p[y] ) {
        ch[x][1]=merge(ch[x][1],y); pushup(x);
        return x;
    } else {
        ch[y][0]=merge(x,ch[y][0]); pushup(y);
        return y;
    }
}

 

平衡树的其他操作的实现

插入新节点(insert)

先把树按照新结点的权值分离,然后再把新结点当作一棵树与两个分离出来的树合并。

 

inline int neo(int v)
{
    val[++tot]=v;
    siz[tot]=1; p[tot]=rand();
    return tot;
}

inline void insert(int v)
{
    int x,y;
    split(root,v,x,y);
    root=merge(merge(x,neo(v)),y);
}

 

删除节点(delete)

按删除节点的权值$v$把树分离成小于$v$,等于$v$,大于$v$三段,然后去掉等于$v$的树的根节点(也就是直接合并根节点的左右子树),然后合并回来。

 

inline void delet(int v)
{
    int x,y,z;
    split(root,v,x,y);
    split(x,v-1,x,z);
    z=merge(ch[z][0],ch[z][1]);
    root=merge(x,merge(z,y));
}

 

查询权值为$k$的节点编号

按照权值分离然后输出左子树大小$+1$即可。

 

inline void kth(int k)
{
    int x,y;
    split(root,k-1,x,y);
    printf("%d\n",siz[x]+1);
    root=merge(x,y);
}

 

查询$k$小值

和其他平衡树一样搜索递归。

 

int rank(int rt,int k)
{
    if( siz[ch[rt][0]]==k-1 ) return val[rt];
    if( siz[ch[rt][0]]>=k ) return rank(ch[rt][0],k);
    return rank(ch[rt][1],k-siz[ch[rt][0]]-1);
}

 

查询前驱后继

按权值分离然后查询分离出的子树的最大/最小值就行了。

 

inline void prev(int v)
{
    int x,y;
    split(root,v-1,x,y);
    printf("%d\n",rank(x,siz[x]));
    root=merge(x,y);
}

inline void nex(int v)
{
    int x,y;
    split(root,v,x,y);
    printf("%d\n",rank(y,1));
    root=merge(x,y);
}

 

 

完整模板

洛谷模板题

//It is made by HolseLee on 27th Sep 2018
//Luogu.org P3369
#include<bits/stdc++.h>
using namespace std;

const int N=1e5+7;
int n,root,tot;
int val[N],ch[N][2],p[N],siz[N];

inline int read()
{
    char ch=getchar(); int num=0; bool flag=false;
    while( ch<0 || ch>9 ) {
        if( ch==- ) flag=true; ch=getchar();
    }
    while( ch>=0 && ch<=9 ) {
        num=num*10+ch-0; ch=getchar();
    }
    return flag ? -num : num;
}

inline void pushup(int rt)
{
    siz[rt]=siz[ch[rt][0]]+siz[ch[rt][1]]+1;
}

void split(int rt,int k,int &x,int &y)
{
    if( !rt ) x=y=0;
    else {
        if( val[rt]<=k ) x=rt, split(ch[rt][1],k,ch[rt][1],y);
        else y=rt, split(ch[rt][0],k,x,ch[rt][0]);
        pushup(rt);
    }
}

int merge(int x,int y)
{
    if( !x || !y ) return x+y;
    if( p[x]<p[y] ) {
        ch[x][1]=merge(ch[x][1],y); pushup(x);
        return x;
    } else {
        ch[y][0]=merge(x,ch[y][0]); pushup(y);
        return y;
    }
}

inline int neo(int v)
{
    val[++tot]=v;
    siz[tot]=1; p[tot]=rand();
    return tot;
}

inline void insert(int v)
{
    int x,y;
    split(root,v,x,y);
    root=merge(merge(x,neo(v)),y);
}

inline void delet(int v)
{
    int x,y,z;
    split(root,v,x,y);
    split(x,v-1,x,z);
    z=merge(ch[z][0],ch[z][1]);
    root=merge(x,merge(z,y));
}

inline void kth(int k)
{
    int x,y;
    split(root,k-1,x,y);
    printf("%d\n",siz[x]+1);
    root=merge(x,y);
}

int rank(int rt,int k)
{
    if( siz[ch[rt][0]]==k-1 ) return val[rt];
    if( siz[ch[rt][0]]>=k ) return rank(ch[rt][0],k);
    return rank(ch[rt][1],k-siz[ch[rt][0]]-1);
}

inline void prev(int v)
{
    int x,y;
    split(root,v-1,x,y);
    printf("%d\n",rank(x,siz[x]));
    root=merge(x,y);
}

inline void nex(int v)
{
    int x,y;
    split(root,v,x,y);
    printf("%d\n",rank(y,1));
    root=merge(x,y);
}

int main()
{
    srand(19260817);
    n=read(); int opt;
    for(int i=1; i<=n; ++i) {
        opt=read();
        if( opt==1 ) insert(read());
        else if( opt==2 ) delet(read());
        else if( opt==3 ) kth(read());
        else if( opt==4 ) printf("%d\n",rank(root,read()));
        else if( opt==5 ) prev(read());
        else nex(read());
    }
    return 0;
}

 

神仙数据结构——FHQ_Treap

原文:https://www.cnblogs.com/cytus/p/9926363.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!