链接,便在今日学习了神仙的fhq_treap。

简介:fhq_treap功能强大,支持splay支持的所有操作,代码简单,仅有两个核心函数$merge$和$split$,更重要的是,它作为无旋的平衡树,是支持可持久化的。

正题:fhq_treap的实现

(0)树的域

对树的每个节点,我们维护 \(ch[2],siz,dat,val\) 分别为左右儿子,子树大小,BST性质的关键字,堆性质的关键字 其中堆是小根堆

(1)split函数

\(void \ split(now,k,\&x,\&y)\)

按照$k$值的意义可以有两种方式理解,其一为把以$now$为根的子树按照$BST$性质的权值按不大于$k$和大于$k$两种分裂成两颗子树,两颗子树的根分别为$x$和$y$。

这个函数是递归执行的,有严格的子问题划分性。

我们可以这样描述一次对整棵树的操作:

比较当前根节点权值$dat[now]$与$k$值 若大于,则根节点和右子树被划分为$y$树,当前根为$now$,$y$树的左子树不确定,进入左子树求解子问题 若不大于,则根节点和左子树被划分为$x$树,当前根为$now$,$x$树的右子树不确定,进入右子树求解子问题

复杂度分析:进入节点所构成的路径相当于对平衡树做了一次查询值为$k$的点的操作,复杂度与树的深度相关,根据$treap$的随机性,可以认为是$O(logn)$

代码实现:

void split(int now,int k,int &x,int &y)//把小于等于k的树分在x上
{
    if(!now) {x=y=0;return;}
    if(dat[now]>k) y=now,split(ls,k,x,ls);
    else x=now,split(rs,k,rs,y);
    updata(now);
}

其中,\(ls\),$rs$为宏定义的左右儿子,$updata$为更新节点的大小$siz$

另一种$k$值的意义为树中排名为$k$的节点,实现起来差不多

(2)merge函数

\(int \ merge(int \ x,int \ y)\)

意义为:把以$x$为根的子树和以$y$值为根的子树合并,返回新根节点。

注意这里有要求:以$x$值为根的子树的BST性质的最大节点小于以$y$为根的子树的最小节点。也就是说,这两棵子树本来就是在BST是一颗完全小于另外一颗,我们只需要在合并时保证堆性质即可。

这个函数是仍然递归执行的,有严格的子问题划分性。

我们可以这样描述一次对整棵树的操作: 比较两个根节点的$val$ 若$val[x]<val[y]$,好的当前根就是$x$了,$x$的左子树不用管了,我们进入$x$的右子树和$y$继续玩 否则,差不多啊反过来就行了

复杂度分析:两颗子树你一下我一下的走到了底,复杂度与子树的深度之和相关,可以认为是$O(logn)$

代码实现:

int Merge(int x,int y)//左树的权值小于右树
{
    if(!x||!y) return x+y;
    if(val[x]<val[y])
    {
        ch[x][1]=Merge(ch[x][1],y);
        updata(x);
        return x;
    }
    else
    {
        ch[y][0]=Merge(x,ch[y][0]);
        updata(y);
        return y;
    }
}

(3)其他常见操作

有了功能强大的$split$和$merge$,我们就可以很轻松的执行其他操作了

\(insert(k)\)

实现:把树按$k$分裂成两个,新建一个点再合并回去

代码:

void Insert(int k)
{
    int x,y;
    split(root,k,x,y);
    root=Merge(Merge(x,New(k)),y);
}

\(extrack(k)\)//删除

实现:把树先分裂成两个,再把含$k$的那一颗把$k$单独分出来。注意如果有重复元素,单独的$k$构成的子树只删除根节点就够了。最后合并回去。

代码:

void extrack(int k)
{
    int x,y,z;
    split(root,k,x,y);
    split(x,k-1,x,z);
    z=Merge(ch[z][0],ch[z][1]);
    root=Merge(x,Merge(z,y));
}

洛谷P3369普通平衡树

Code:

#include <cstdio>
#include <cstdlib>
#define ls ch[now][0]
#define rs ch[now][1]
const int N=100010;
int siz[N],ch[N][2],dat[N],val[N],tot,root;
void updata(int now)
{
    siz[now]=siz[ls]+siz[rs]+1;
}
void split(int now,int k,int &x,int &y)//把小于等于k的树分在x上
{
    if(!now) {x=y=0;return;}
    if(dat[now]>k) y=now,split(ls,k,x,ls);
    else x=now,split(rs,k,rs,y);
    updata(now);
}
int Merge(int x,int y)//左树的权值小于右树
{
    if(!x||!y) return x+y;
    if(val[x]<val[y])
    {
        ch[x][1]=Merge(ch[x][1],y);
        updata(x);
        return x;
    }
    else
    {
        ch[y][0]=Merge(x,ch[y][0]);
        updata(y);
        return y;
    }
}
int New(int k)
{
    siz[++tot]=1;dat[tot]=k;val[tot]=rand();
    return tot;
}
void Insert(int k)
{
    int x,y;
    split(root,k,x,y);
    root=Merge(Merge(x,New(k)),y);
}
void extrack(int k)
{
    int x,y,z;
    split(root,k,x,y);
    split(x,k-1,x,z);
    z=Merge(ch[z][0],ch[z][1]);
    root=Merge(x,Merge(z,y));
}
void Rank(int k)
{
    int x,y;
    split(root,k-1,x,y);
    printf("%d\n",siz[x]+1);
    root=Merge(x,y);
}
void frank(int now,int x)
{
    while(233)
    {
        if(siz[ls]>=x) now=ls;
        else if(siz[ls]+1<x) x-=siz[ls]+1,now=rs;
        else {printf("%d\n",dat[now]);return;}
    }
}
void pre(int k)
{
    int x,y;
    split(root,k-1,x,y);
    frank(x,siz[x]);
    root=Merge(x,y);
}
void suc(int k)
{
    int x,y;
    split(root,k,x,y);
    frank(y,1);
    root=Merge(x,y);
}
int main()
{
    int t,opt,x;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&opt,&x);
        if(opt==1) Insert(x);
        else if(opt==2) extrack(x);
        else if(opt==3) Rank(x);
        else if(opt==4) frank(root,x);
        else if(opt==5) pre(x);
        else suc(x);
    }
    return 0;
}



2018.7.28

相关文章:

  • 2021-06-06
  • 2021-10-21
  • 2022-12-23
  • 2022-12-23
  • 2021-12-24
  • 2022-12-23
  • 2021-08-08
猜你喜欢
  • 2021-09-29
  • 2021-06-27
  • 2022-01-25
  • 2021-12-22
  • 2021-09-23
  • 2022-01-02
  • 2022-12-23
相关资源
相似解决方案