利用其他人其中考试的时间,终于学完了非旋转Treap,它与普通Treap的区别就是它不旋转废话。前置知识只有BST和可并堆。

BST看这个博客,解释的挺清楚的。https://www.cnblogs.com/jiangminghong/p/9999884.html

可并堆就是用很快的时间合并两个堆。如果裸上一个并查集的话就是nlog2n.这个复杂度我们是不能接受的。正常的可并堆是维护一棵左偏树,我们用一个参数dis[x]表示从x点出发能够向右走的最大步数。每次两个堆合并时,我们就把一个堆扔到另一个堆里就行了。

在旋转Treap中,我们用随机数来保持平衡树的平衡.非旋转Treap也是同样的做法。

1 int lson[N];//表示当前点的左儿子
2 int rson[N];//表示当前点的右儿子
3 int key[N];//表示当前点的随机数(保持平衡树的平衡)
4 int val[N];//表示当前点的权值
5 int size[N];//表示当前点子树的大小

这就是我们需要维护的信息。

看看操作吧

 1.Merge【合并】O(logn)
 2.Split【拆分】O(logn)

只有两个

可资瓷使用范围

1.Insert Newnode+Merge O(logn)

2.Delete Split+Split+Merge O(logn)

3.Find_kth Split+Split O(logn)

4.Query  Split+Split  O(logn)

5.各种区间操作

1.split

split操作就是以一个点为界限,将平衡树分裂成两棵平衡树,保证左边平衡树中任意点权值小于右边平衡树中任意点的权值。

以将平衡树分裂成权值<=val和权值>val两部分为例:从根节点开始往下查找,当当前点权值<=val时,将当前点及它的右子树接到分裂后第二棵平衡树的左子树上;反之则将当前点及它的左子树接到分裂后第一棵平衡树的右子树上,直到找到叶子节点为止。

void split(int now,int k,int &x,int &y)//now表示现在的位置,k表示要查询的权值,x,y分别为拆分后两子树的根
{
    if(!now)
    {
        x=y=0;
        return ;
    }
    if(val[now]<=k)
        x=now,split(rson[now],k,rson[now],y);
    else
        y=now,split(lson[now],k,x,lson[now]);
    push_up(now);
    return ;
}

如果想要拆分成前K个点与后n-K个点也是同理的,判断的条件就是size喽!

void split(int now,int k,int &x,int &y)
{
    if(!now)
    {
        x=y=0;
        return ;
    }
    if(size[lson[now]]<k)
        x=now,split(rson[now],k-size[lson[now]]-1,rson[now],y);
    else
        y=now,split(lson[now],k,x,lson[now]);
    push_up(now);
}

2.merge

 merge这个操作就是将两个堆合并,我们请出可并堆。可并堆的合并方式就是按照随机数大小来合并。

int merge(int x,int y)
{
    if(!x||!y)
        return x|y;
    if(key[x]<key[y])
    {
        rson[x]=merge(rson[x],y);
        push_up(x);
        return x;
    }
    else
    {
        lson[y]=merge(x,lson[y]);
        push_up(y)
        return y;
    }
}

有木有一直好奇push_up函数啊,其实和线段树一样,维护子树的一些信息。

有了这两个sao操作,是不是就可以在序列上想干嘛就干嘛了?

1.Newnode

新建节点,加入平衡树

int newnode(int x)
{
    size[++tot]=1;
    val[tot]=x;
    key[tot]=rand()*rand();
    return tot;
}


split(root,a,x,y);
root=merge(merge(x,newnode(a)),y);

2.Delete

划分成3个区间,之后合并

split(root,a,x,z);
split(x,a-1,x,y);
y=merge(lson[y],rson[y]);
root=merge(merge(x,y),z);

3.Query(查询以x为根排名第K的数)

判断K和左子树大小的关系,根据size找到答案。

int kth(int now,int k)
{
    while(1)
    {
        if(now==0)
            break;
        if(k<=size[lson[now]])
            now=lson[now];
        else if(k==size[lson[now]])
            return now;
        else
            k-=size[lson[now]]+1,now=rson[now];
    }
    return 0;
}

4.前驱&后继

split(root,a-1,x,y); 
printf("%d\n",val[kth(x,size[x])]);
root=merge(x,y);
//前驱

split(root,a,x,y);
printf("%d\n",val[kth(y,1)]);
root=merge(x,y);
//后继

LuoguP3369 普通平衡树

#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define N 1000100
int tot;
int size[N];
int lson[N];
int rson[N];
int key[N];
int val[N];
int n,m;
int root;
int x,y;
void push_up(int x){size[x]=size[lson[x]]+size[rson[x]]+1;return ;}
int newnode(int x)
{
    size[++tot]=1;
    val[tot]=x;
    key[tot]=rand();
    return tot;
}
int merge(int x,int y)
{
    if(!x||!y)
        return x+y;
    if(key[x]<key[y])
    {
        rson[x]=merge(rson[x],y);
        push_up(x);
        return x;
    }
    else
    {
        lson[y]=merge(x,lson[y]);
        push_up(y);
        return y;
    }
}
void split(int now,int k,int &x,int &y)
{
    if(!now)
        x=y=0;
    else
    {
        if(val[now]<=k)
            x=now,split(rson[now],k,rson[now],y);
        else
            y=now,split(lson[now],k,x,lson[now]);
        push_up(now);
    }
    return ;
}
int kth(int now,int k)
{
    while(1)
    {
        if(k<=size[lson[now]])
            now=lson[now];
        else if(k==size[lson[now]]+1)
            return now;
        else
            k-=size[lson[now]]+1,now=rson[now];
    }
    return 0;
}
int main()
{
    srand(20030305);
    scanf("%d",&n);
    int opt,a;
    while(n--)
    {
        scanf("%d%d",&opt,&a);
        if(opt==1)
        {
            split(root,a,x,y);
            root=merge(merge(x,newnode(a)),y);
        }
        if(opt==2)
        {
            int z;
            split(root,a,x,z);
            split(x,a-1,x,y);
            y=merge(lson[y],rson[y]);
            root=merge(merge(x,y),z);
        }
        if(opt==3)
        {
            split(root,a-1,x,y);
            printf("%d\n",size[x]+1);
            root=merge(x,y);
        }
        if(opt==4)
            printf("%d\n",val[kth(root,a)]);
        if(opt==5)
        {
            split(root,a-1,x,y); 
            printf("%d\n",val[kth(x,size[x])]);
            root=merge(x,y);
        }
        if(opt==6)
        {
            split(root,a,x,y);
            printf("%d\n",val[kth(y,1)]);
            root=merge(x,y);
        }
    }
    return 0;
}

 

版权声明:本文为342zhuyongqi原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://www.cnblogs.com/342zhuyongqi/p/10028990.html