Treap

treap
[详解](https://www.cnblogs.com/fusiwei/p/12884254.html)

treap

这里的treap指有旋treap

如何种一棵(香蕉树)有旋treap

Treap=Tree+heap。
所以Treap树堆其实就是树+堆。树是二叉查找树BST,堆是二叉堆,大根堆小根堆都可以。
树堆既是一棵二叉查找树,也是一个二叉堆。但是这两种数据结构貌似还是矛盾的存在,如果是二叉查找树,就不能是一个堆,如果是一个堆,那么必然不是二叉查找树。
所以树堆用了一个很巧妙的方式解决这个问题:给每个键值一个随机附加的优先级,让键值满足二叉查找树的结构,让优先级满足二叉堆的结构

树堆实现平衡树的特点

BST中讲过,普通的BST具有很强的不确定性,如果数据特殊,建树的时候可能直接变成一条链。
不仅如此,插入删除的时候也很麻烦。因为如果插入或者删除,整个树原来的结构就会被打乱,这会为遍历和查找带来灾难性的后果。
所以我们推出了平衡树。就是通过将树旋转来动态维护这个树形态是平衡的,这样查找的复杂度就是O(log)级别的,是一种稳定的复杂度。
树堆是一种平衡树,它通过为键值(也就是我们需要维护成BST的)赋予优先级,使之也满足堆结构来进行旋转,成为一棵平衡树。
但是我们需要注意一点:树堆的优先级是随机赋予的。也就是说,这个数据结构其实是一个随机化的数据结构。
但是,这不是树堆的缺点,因为只有随机化赋予优先级,才有可能保证树堆的复杂度是O(log)的级别。
那么,上述性质也说明了,树堆并不是一个规则形态的二叉树,更不是堆需要满足的完全二叉树。
甚至它也不符合平衡树的定义:每个节点左右子树高度相差≤1,所以我们说树堆是近似实现平衡。
但是通过形态定义二叉树的方式并不绝对。我们换一种方式来对平衡树进行定义:能够保证时间复杂度的BST,就是平衡树。

树堆的操作

首先当我们理解Treap的操作的时候,需要先对旋转这个事情有一个大体的定义。
上图:

光看这个箭头的话,还是很容易理解什么是旋转的。
但是可能给读者造成困扰的是:这个B节点的父亲怎么变了?
原因是这样的:我们在进行旋转操作的时候,要保证BST的节点遍历顺序是一样的,而BST的节点遍历顺序是中序遍历(这个是按BST的定义来的),也就是说只有这样才能保证遍历序不变的情况下调换节点位置。
所以,针对这个图,我们把“P/F”看成一对节点,就能很好地理解这个“左/右旋”的操作了。

Treap的插入

首先我们了解一下BST的插入方式。
其实很简单,就是一个新节点插进去,从根节点开始不停地与当前节点比大小,一直到这个节点成为叶子节点为止。
因为Treap要同时维护BST和堆,所以我们还需要在里面加上旋转操作。
如果是右儿子的话要左旋,左儿子要右旋(旋转操作请结合上图理解)。
然后我们又多统计了一个size的数据,这个数据表示子树大小,在统计当前数x是第几大的时候会很方便。
上代码:

struct node
{
    int val,pri,size,lson,rson;//val键值,pri优先级,size子树大小。
}tree[maxn];
int tot;
void L_rotate(int &pos)
{
    node x=tree[tree[pos].rson];//x表示要转到上面的节点
    tree[pos].rson=x.lson;
    x.lson=pos;
    x.size=tree[pos].size;
    maintain(pos);
    tree[pos]=x;
}
void R_rotate(int &pos)
{
    node x=tree[tree[pos].lson];
    tree[pos].lson=x.rson;
    x.size=tree[pos].size;
    maintain(pos);
    tree[pos]=x;
}
void maintain(int pos)
{
    tree[pos].size=tree[tree[pos].lson].size+tree[tree[pos].rson].size+1;
}
void insert(int &pos,int x)
{
    if(!pos)
    {
        pos=++tot;
        tree[pos].val=x;
        tree[pos].pri=rand();
    }
    if(x<tree[pos].val)
    {
        insert(tree[pos].lson,x);
        if(tree[pos].pri<tree[lson].pri)//如果优先级不匹配,就旋转(这里维护的是大根堆)
            R_rotate(pos);
    }
    else
    {
        insert(tree[pos].rson,x);
        if(tree[pos].pri<tree[rson].pri)
            L_rotate(pos);
    }
    maintain(pos);
}

Treap的删除

删除操作的大体思路和插入是一样的。
也是要保证删除前后满足Treap的双重结构。
大致是这样的思路:

  • 首先找到这个点在哪。
  • 然后
    • 如果这个点已经是叶子节点,就直接将其删除
    • 如果不是,就一层层地将它转到底部,然后进行删除。

这和我们的插入操作有异曲同工之妙,就是进行操作的一定是叶子节点,如果不是叶子的话是不能粗暴删除的。需要转。

void remove(int &pos,int x)
{
    if(!pos)
        return;
    if(x==tree[pos].val)
    {
        if(tree[pos].lson|tree[pos].rson)//如果非叶子节点
        {
            if(tree[tree[pos].lson].pri>tree[tree[pos].rson].pri)
                R_rotate(pos),remove(tree[pos].lson,x);
            else
                L_rotate(pos),remove(tree[pos].rson,x);
            maintain(pos);
        }
        else
            pos=0;
    }
    else
    {
        if(x<tree[pos].val)
            remove(tree[pos].lson,x);
        else
            remove(tree[pos].rson,x);
    }
    if(pos)
        maintain(pos);
}

Treap的查询(某数排名)

查询操作不涉及修改,根据树堆的双重性质,其操作是跟BST是一样的。

int rank(int pos,int x)
{
    if(!pos)
        return 1;
    if(x<tree[pos].val)
        return rank(tree[pos].lson,x);
    else
        return rank(tree[pos].rson,x)+tree[tree[pos].lson].size+1;
}
int rank(int pos,int x,int &cnt)
{
    if(!pos)
        return -1;
    if(x==tree[pos].val)
        return cnt+=tree[tree[pos].lson].size;
    else
    {
        if(x<tree[pos].val)
            rank(tree[pos].lson,x,cnt);
        else
            cnt+=(tree[tree[pos].lson].size+1),rank(tree[pos].rson,x,cnt);
    }
}

Treap的查询(第k大)

int kth(int pos,int k)
{
    if(!pos || k<=0 || k>tree[pos].size)
        return -1;
    if(k==tree[tree[pos].lson].size+1)
        return tree[pos].val;
    else if(k<=tree[tree[pos].lson].size)
        return kth(tree[pos].lson,k);
    else
        return kth(tree[pos].rson,k-tree[tree[pos].lson].size-1);
}

Treap的查询(前驱/后继)

#define INF 1e9
int prev(int pos,int x)
{
    if(!pos)
        return -INF;
    if(x<tree[pos].val)
        return prev(tree[pos].lson,x);
    else
        return max(tree[pos].val,prev(tree[pos].rson,x));
}
int nxt(int pos,int x)
{
    if(!pos)
        return INF;
    if(x>=tree[pos].val)
        return nxt(tree[pos].rson,x);
    else
        return min(tree[pos].val,nxt(tree[pos].lson,x));
}

Treap的遍历

Treap的遍历是中序遍历,遵循先左后右的原则。

void dfs(int pos)
{
    if(tree[pos].lson)
        dfs(tree[pos].lson);
    printf("%d ",tree[pos].val);
    if(tree[pos].rson)
        dfs(tree[pos].rson);
}

posted @ 2024-09-13 19:14  yzc_is_SadBee  阅读(4)  评论(0编辑  收藏  举报