首页 > > 程序设计 > C/C++ >

平衡树Treap模板与原理

2018-06-17 21:23:13来源：未知阅读 ()

这次我们来讲一讲Treap(splay以后再更)

平衡树是一种排序二叉树(或二叉搜索树)，所以排序二叉树可以迅速地判断两个值的大小，当然操作肯定不止那么多(不然我们还学什么)。

而平衡树在排序二叉树的基础上主要是增加了一个优化：就是高度较为平衡，并可以动态平衡。

而今天要讲的treap就是一种动态平衡的方法。

首先说声抱歉，因为没有那么多的时间，所以无法把左旋和右旋两种操作具体的讲，但是可以看刘汝佳的蓝书，讲得还是挺清楚的。

现在开始。

treap用的是一种比较玄学的方法，就是将每一个点还附上一个随机值，然后按照堆的性质，不管大小根堆都是一样的，所以我们每加入一个值，就利用上浮操作进行旋转，保持堆的性质，且不改变排序二叉树的性质。

操作很好理解，就是每次insert时进行判断就行了，若不满足，则上浮。

具体的排序二叉树的操作，我就不再赘述了，如果要学，可以看刘汝佳的蓝书(感觉在跟刘汝佳打广告)。

下面送上代码。

  1 #include<iostream>
  2 #include<cstdio>
  3 #include<cstring>
  4 #include<algorithm>
  5 #include<ctime>
  6 using namespace std;
  7 int n,root,size,ans;
  8 struct P{
  9     int l,r,sz,key,rd,re;//re为重复次数，key为加入权值
 10 }t[1000005];
 11 void update(int k)
 12 {
 13     t[k].sz=t[t[k].l].sz+t[t[k].r].sz+t[k].re;
 14 }
 15 void left(int &k)//右旋
 16 {
 17     int y=t[k].r;
 18     t[k].r=t[y].l;
 19     t[y].l=k;
 20     t[y].sz=t[k].sz;
 21     update(k);
 22     k=y;
 23 }
 24 void right(int &k)//左旋
 25 {
 26     int y=t[k].l;
 27     t[k].l=t[y].r;
 28     t[y].r=k;
 29     t[y].sz=t[k].sz;
 30     update(k);
 31     k=y;
 32 }
 33 void init(int &k,int x)//加入操作
 34 {
 35     if(k==0)
 36     {
 37         size++;
 38         k=size;
 39         t[k].sz=1;
 40         t[k].re=1;
 41         t[k].key=x;
 42         t[k].rd=rand();
 43         return;
 44     }
 45     t[k].sz++;
 46     if(t[k].key==x)  t[k].re++;
 47     else{
 48         if(x>t[k].key)
 49         {
 50             init(t[k].r,x);
 51             if(t[t[k].r].rd<t[k].rd)   left(k);//这里和下面都是判断是不是满足堆的性质
 52         }
 53         if(x<t[k].key)
 54         {
 55             init(t[k].l,x);
 56             if(t[t[k].l].rd<t[k].rd)   right(k);
 57         }
 58     }
 59 } 
 60 void del(int &k,int x)
 61 {
 62     if(k==0)  return;
 63     if(t[k].key==x)
 64     {
 65         if(t[k].re>1)
 66         {
 67             t[k].re--;
 68             t[k].sz--;
 69             return;
 70         }
 71         if(t[k].l*t[k].r==0)  k=t[k].l+t[k].r;//k代表指针的移动，所以移动到了左或右儿子
 72         else
 73         {
 74             if(t[t[k].l].rd<t[t[k].r].rd){
 75                 right(k);
 76                 del(k,x);
 77             }
 78             else{
 79                 left(k);
 80                 del(k,x);
 81             }
 82         }
 83     }
 84     else{
 85         if(x>t[k].key)
 86         {
 87             t[k].sz--;
 88             del(t[k].r,x);
 89         }
 90         else{
 91             t[k].sz--;
 92             del(t[k].l,x);
 93         }
 94     }
 95 }
 96 int rank1(int k,int x)//找x的排名
 97 {
 98     if(k==0)  return 0;
 99     if(t[k].key==x)  return t[t[k].l].sz+1;
100     if(x>t[k].key)   return t[t[k].l].sz+rank1(t[k].r,x)+t[k].re;
101     if(x<=t[k].key)  return    rank1(t[k].l,x);
102 }
103 int rank2(int k,int x)//找排名为x的数
104 {
105     if(k==0)  return 0;
106     else if(x<=t[t[k].l].sz)    return rank2(t[k].l,x);
107     else if(x>t[t[k].l].sz+t[k].re)  return rank2(t[k].r,x-t[t[k].l].sz-t[k].re);
108     else  return  t[k].key; 
109 }
110 void pre(int k,int x)//找前驱
111 {
112     if(k==0)  return;
113     if(t[k].key<x)
114     {
115          ans=k;
116          pre(t[k].r,x);
117     }
118     else pre(t[k].l,x);
119 }
120 void nxt(int k,int x)//找后继
121 {
122     if(k==0)  return;
123     if(t[k].key>x)
124     {
125         ans=k;
126         nxt(t[k].l,x);
127     }
128     else nxt(t[k].r,x);
129 }
130 int main()
131 {
132     srand(time(0));
133     scanf("%d",&n);
134     for(int i=1;i<=n;i++)
135     {
136         int num,x;
137         scanf("%d%d",&num,&x);
138         if(num==1)  init(root,x);
139         if(num==2)  del(root,x);
140         if(num==3)  printf("%d\n",rank1(root,x));
141         if(num==4)  printf("%d\n",rank2(root,x));
142         if(num==5)
143         {
144             pre(root,x);
145             printf("%d\n",t[ans].key);
146         }
147         if(num==6)
148         {
149             nxt(root,x);
150             printf("%d\n",t[ans].key);
151         }
152     }
153     return 0;
154 }