二叉查找树（BST)、平衡二叉树(AVL树)（只有插入说明）

二叉查找树（BST)、平衡二叉树(AVL树)（只有插入说明）

二叉查找树(BST)

　　特殊的二叉树，又称为排序二叉树、二叉搜索树、二叉排序树。

　　二叉查找树实际上是数据域有序的二叉树，即对树上的每个结点，都满足其左子树上所有结点的数据域均小于或等于根结点的数据域，右子树上所有结点的数据域均大于根结点的数据域。如下图所示：

二叉查找树通常包含查找、插入、建树和删除操作。

二叉查找树的创建

对于一棵二叉查找树，其创建与二叉树的创建很类似，略有不同的是，二叉查找树，为了保证整棵树都关于根结点的大小呈左小右大的特征，在创建时，需要根据当前结点的大小来判断插入位置，给出如下代码：

template<typename T>
void  BSTree<T>::createBSTreeByFile(ifstream &f){
    T e;
    queue<BSNode<T>*> q;

    while(!f.eof()){
        InputFromFile(f, e);
        Insert(root, e);
    }
}

template<typename T>
void BSTree<T>::Insert(BSNode<T>* &t, T x){//得用指针的引用，不然传参时由于形参实例化，并不能成功创建二叉树

    if(t==NULL){
        t = new BSNode<T>;
        t->data = x;
        t->lchild = t->rchild = NULL;
        return;
    }

    if(x<=t->data){
        Insert(t->lchild, x);
    }
    else{
        Insert(t->rchild, x);
    }
}

二叉查找树的查找

二叉查找树的查找有递归和非递归两种，对于递归方式，其递归边界为树的终止结点，非递归方式则采取对树中所有结点采取BFS或者DFS进行遍历的方式。

对于非递归方式，给出采取DFS的遍历方式，在这种方式中，通常采用入栈的方式，来访问每个结点，而根据访问的先后顺序，又分为，前序、中序和后序三种遍历方式。以前序遍历为例，通常以根、左、右的顺序访问遍历每个结点，而中序遍历方式，则以左、根、右的顺序遍历，后序则以左右根的顺序来访问。下面给出三种遍历方式的代码：前序遍历：

 1 template<typename T>
 2 void BSTree<T>::PreOrderTraverse(void(*visit)(BSNode<T>&))const{
 3     stack<BSNode<T>*> s;
 4     BSNode<T> *t = root;
 5     while(NULL!=t || !s.empty()){
 6         if(NULL!=t){
 7             s.push(t);
 8             visit(*t);
 9             t = t->lchild;
10         }
11         else{
12             t = s.top();
13             s.pop();
14             t = t->rchild;
15         }
16     }
17     cout<<endl;
18 }

中序遍历：

 1 template<typename T>
 2 void BSTree<T>::InOrderTraverse(void(*visit)(BSNode<T>&))const{
 3     stack<BSNode<T>*> s;
 4     BSNode<T> *q;
 5
 6     q = root;
 7
 8     while(!s.empty()||q!=NULL){
 9         if(q!=NULL){
10             s.push(q);
11             q = q->lchild;
12         }
13         else{
14             q = s.top();
15             s.pop();
16             visit(*q);
17             q = q->rchild;
18         }
19     }
20     cout<<endl;
21 }

后序遍历，对于后序遍历，直接采用入栈的方式进行访问，是不行的，因为根结点被访问两次，无法保证你在弹栈后，对该结点如何操作，因此，需要另设置一个flag参数，来指明该节点是否左右子树都访问过，代码如下，我这里是令定义一个结构体，来实现：

/*结构体部分*/
enum Tags{Left, Right};

template<typename T>struct StackElem
{
    BSNode<T> *p;
    Tags flag;
};

/*后序遍历代码部分*/
template<typename T>
void BSTree<T>::PostOrderTraverse(void(*visit)(BSNode<T>&))const{
    StackElem<T> se;
    stack<StackElem<T> > s;

    BSNode<T> *t;
    t = root;

    if(t==NULL){
        return;
    }
    while(t!=NULL||!s.empty()){
        while(t!=NULL){
            se.flag = Left;
            se.p = t;
            s.push(se);
            t = t->lchild;
        }
        se = s.top();
        s.pop();
        t = se.p;
        if(se.flag==Left){
            se.flag = Right;
            s.push(se);
            t = t->rchild;
        }
        else{
            visit(*t);
            t = NULL;
        }
    }
}

以下是递归实现部分，递归实现，则是以二叉树边界为递归边界，前面已经说过了，其余逻辑与非递归一致，因为递归的过程，可以看作是一个入栈和弹栈的过程，即，在未到达边界时，通过递归，来访问下一个结点，例如左结点，当触及边界，则访问该结点，由于每次递归状态都被计算机保存，因此，在访问一个结点以后，返回上一个结点的状态，会依次访问上去。

递归前序遍历：

 1 template<typename T>
 2 void BSTree<T>::PreTraverse(BSNode<T> *t, void(*visit)(BSNode<T>&))const{
 3     if(t==NULL){
 4         return;
 5     }
 6     else{
 7         visit(*t);
 8         PreTraverse(t->lchild, visit);
 9         PreTraverse(t->rchild, visit);
10     }
11 }

递归中序遍历：

 1 template<typename T>
 2 void BSTree<T>::InTraverse(BSNode<T> *t, void(*visit)(BSNode<T>&))const{
 3     if(t==NULL){
 4         return;
 5     }
 6     else{
 7         InTraverse(t->lchild, visit);
 8         visit(*t);
 9         InTraverse(t->rchild, visit);
10     }
11 }

递归后序遍历：

1 template<typename T>
2 void BSTree<T>::PostTraverse(BSNode<T> *t, void(*visit)(BSNode<T>&))const{
3     if(t!=NULL){
4         PostTraverse(t->lchild, visit);
5         PostTraverse(t->rchild, visit);
6         visit(*t);
7     }
8 }

平衡二叉树（AVL树）

　　平衡二叉树是由前苏联的两位数学家G.M.Adelse-Velskil和E.M.Landis提出，因此一般也称作AVL树，AVL树本质还是一棵二叉查找树，只是在其基础上增加了“平衡”的要求。所谓平衡是指，对AVL树的任意结点来说，其左子树与右子树的高度之差的绝对值不超过1，其中左子树与右子树的高度因子之差称为平衡因子。

　　如下所示，就是一棵由{1，2，3，4，5，7，8}构建的AVL树：

　　

　　只要能随时保证每个结点平衡因子的绝对值不超过1，AVL的高度就始终能保持O(logn)级别，由于需要对每个结点都得到平衡因子，因此需要在树的结构中加入一个变量height来记录以当前结点为根结点的子树的高度。

AVL树的创建

　　AVL树的创建是基于二叉查找树的插入代码的基础上，增加平衡操作的。需要从插入的结点开始从下往上判断结点是否失衡，因此，需要在调用insert函数以后，更新当前子树的高度，并在这之后根据树型来进行相应的平衡操作。那么，怎么进行平衡操作呢？AVL树的插入是需要采取左旋或者右旋操作的，即，插入后，由于插入操作，导致某棵子树的高度超过了另一棵子树高度的2个结点高度，这样就破坏了树的平衡性，需要做出调整。

右旋操作

如下所示一棵简单的AVL树，对其进行插入操作以后：

一棵简单的AVL树

变成了下图这样的AVL树：

这样子就失衡了，所谓右旋操作，就是将这棵AVL树，从最靠近插入结点的失衡结点处，通过往右子树调整，使整棵树的每个结点的平衡因子变为正常，不如上图的树，离插入节点3最近的失衡结点是7，

则可以通过下图所示的操作，来平衡二叉树，即调整整棵树平衡因子：

同样，左旋也与此类似。但是，如果5结点本身就有右结点，即如下所示：

这样，在经过右旋操作以后，这棵树还是不平衡的，旋转后这棵树如下所示：

因此，还需要进行一次旋转，显然，继续右旋已经无法满足我们的需求，那么要如何进行操作，才能使这棵树回复平衡呢？（在后续中，会进行讲解）

左旋操作

左旋操作与右旋操作是类似的，都属于对子树的单旋转。

左旋与右旋一样，同样也存在这样的问题，如果该树的右子树的左结点存在，则单一通过左旋是做不到的，那么应该如何处理呢？

其实，以L和R来表示，插入结点的位置，有以下四种情况：

从上表可以看出，左旋和右旋两种情况中，左右结点若存在的话，就是上表中的RL和LR情况。则，只需要对两种情况分别按照上表采取相应的操作就可以解决，如下图所示：

LR型

RL型

由此，就能实现AVL树的平衡，下面给出代码：

AVLTree.h

  1 #ifndef _AVLTREE_H_
  2 #define _AVLTREE_H_
  3 #include "C.h"
  4 #include "AVLNode.h"
  5 #include "Function.h"
  6
  7 typedef int T;
  8
  9 using namespace std;
 10
 11 template<typename T>
 12 class AVLTree{
 13 private:
 14     AVLNode<T> *root;
 15     Destroy(AVLNode<T> *t){
 16         if(t!=NULL){
 17             Destroy(t->lchild);
 18             Destroy(t->rchild);
 19             delete t;
 20             t = NULL;
 21         }
 22         return 0;
 23     }
 24 public:
 25     AVLTree(){
 26         root = NULL;
 27     }
 28     ~AVLTree(){
 29         Destroy(root);
 30     }
 31
 32     AVLNode<T>* newAVLNode(T x);    //创建新结点
 33     void Insert(AVLNode<T>* &t, T x);
 34     void createAVLTreeFromFile(ifstream &f);
 35     AVLNode<T>* Root()const;
 36     int AVLTreeDepth(AVLNode<T> *t)const;
 37     int getAVLTreeHeight(AVLNode<T>* t)const;   //获取当前结点的高度
 38     int getBalanceFactor(AVLNode<T>* t)const;   //计算当前结点的高度
 39     void updateAVLNodeHeight(AVLNode<T>* &t);
 40     T getElem(AVLNode<T>* t)const;
 41     bool getElemExist(AVLNode<T>* &t)const;
 42     void LeftRotation(AVLNode<T>* &t);
 43     void RightRotation(AVLNode<T>* &t);
 44     void PreOrderTraverse(AVLNode<T>* t, void(*visit)(AVLNode<T>&))const;
 45     void PostOrderTraverse(AVLNode<T>* t, void(*visit)(AVLNode<T>&))const;
 46 };
 47
 48 template<typename T>
 49 AVLNode<T>* AVLTree<T>::newAVLNode(T x){
 50     AVLNode<T>* avlnode = new AVLNode<T>;
 51     avlnode->data = x;
 52     avlnode->height = 1;
 53     avlnode->lchild = avlnode->rchild = NULL;
 54     return avlnode;
 55 }
 56
 57 template<typename T>
 58 void AVLTree<T>::Insert(AVLNode<T>* &t, T x){
 59     if(t==NULL){
 60         t = newAVLNode(x);
 61         return;
 62     }
 63     if(x==t->data){//结点已经存在，直接返回
 64         return;
 65     }
 66     if(x < t->data){
 67         Insert(t->lchild, x);
 68         updateAVLNodeHeight(t);
 69         if(getBalanceFactor(t)==2){
 70             if(getBalanceFactor(t->lchild)==1){
 71                 RightRotation(t);
 72             }
 73             else if(getBalanceFactor(t->lchild)==-1){
 74                 LeftRotation(t->lchild);
 75                 RightRotation(t);
 76             }
 77         }
 78     }
 79     else{
 80         Insert(t->rchild, x);   //值比当前结点大，往右子树插入
 81         updateAVLNodeHeight(t);     //更新树高
 82         if(getBalanceFactor(t)==-2){
 83             if(getBalanceFactor(t->rchild)==-1){ //RR型
 84                 LeftRotation(t);
 85             }
 86             else if(getBalanceFactor(t->rchild)==1){
 87                 RightRotation(t->rchild);
 88                 LeftRotation(t);
 89             }
 90         }
 91     }
 92 }
 93
 94 template<typename T>
 95 void AVLTree<T>::createAVLTreeFromFile(ifstream &f){
 96     T e;
 97     while(!f.eof()){
 98         InputFromFile(f, e);
 99         Insert(root, e);
100     }
101 }
102
103 template<typename T>
104 AVLNode<T>* AVLTree<T>::Root()const{
105     return root;
106 }
107
108 template<typename T>
109 int AVLTree<T>::AVLTreeDepth(AVLNode<T> *t)const{
110     int i,j;
111     if(t==NULL){
112         return 0;
113     }
114     else{
115         i = AVLTreeDepth(t->lchild);
116         j = AVLTreeDepth(t->rchild);
117     }
118     return i>j ? i+1 : j+1;
119 }
120
121 template<typename T>
122 int AVLTree<T>::getAVLTreeHeight(AVLNode<T>* t)const{
123     if(t==NULL){
124         return 0;
125     }
126     return t->height;
127 }
128
129 template<typename T>
130 int AVLTree<T>::getBalanceFactor(AVLNode<T>* t)const{
131     if(t==NULL){
132         return 0;
133     }
134     return getAVLTreeHeight(t->lchild) - getAVLTreeHeight(t->rchild);
135 }
136
137 template<typename T>
138 void AVLTree<T>::updateAVLNodeHeight(AVLNode<T>* &t){
139     t->height = max(getAVLTreeHeight(t->lchild), getAVLTreeHeight(t->rchild)) + 1;
140 }
141
142 template<typename T>
143 T AVLTree<T>::getElem(AVLNode<T>* t)const{
144     return t->data;
145 }
146
147 template<typename T>
148 bool AVLTree<T>::getElemExist(AVLNode<T>* &t)const{//判断当前结点是否为空
149     if(t!=NULL){
150         return true;
151     }
152     return false;
153 }
154
155 template<typename T>
156 void AVLTree<T>::LeftRotation(AVLNode<T>* &t){
157     AVLNode<T> *temp = t->rchild;
158     t->rchild = temp->lchild;
159     temp->lchild = t;
160     updateAVLNodeHeight(t);
161     updateAVLNodeHeight(temp);
162     t = temp;
163 }
164
165 template<typename T>
166 void AVLTree<T>::RightRotation(AVLNode<T>* &t){
167     AVLNode<T> *temp = t->lchild;
168     t->lchild = temp->rchild;
169     temp->rchild = t;
170     updateAVLNodeHeight(t);
171     updateAVLNodeHeight(temp);
172     t = temp;
173 }
174
175 template<typename T>
176 void AVLTree<T>::PreOrderTraverse(AVLNode<T>* t, void(*visit)(AVLNode<T>&))const{
177     if(t!=NULL){
178         visit(*t);
179         PreOrderTraverse(t->lchild, visit);
180         PreOrderTraverse(t->rchild, visit);
181     }
182 }
183
184 template<typename T>
185 void AVLTree<T>::PostOrderTraverse(AVLNode<T>* t, void(*visit)(AVLNode<T>&))const{
186     if(t!=NULL){
187         PostOrderTraverse(t->lchild, visit);
188         PostOrderTraverse(t->rchild, visit);
189         visit(*t);
190     }
191 }
192 #endif // _AVLTREE_H_

Function.h

 1 #ifndef _FUNCTION_H_
 2 #define _FUNCTION_H_
 3 #include "C.h"
 4 #include "AVLNode.h"
 5 #include "AVLTree.h"
 6
 7 typedef int T;
 8
 9 using namespace std;
10
11 bool InputFromFile(ifstream &f, T &e){
12     f>>e;
13     return f.good();
14 }
15
16 void visit(AVLNode<T> &t){
17     cout<<t.data<<" ";
18 }
19
20 #endif // _FUNCTION_H_

C.h

 1 #ifndef _C_H_
 2 #define _C_H_
 3 #include<iostream>
 4 #include<string>
 5 #include<stdio.h>
 6 #include<algorithm>
 7 #include<map>
 8 #include<math.h>
 9 #include<queue>
10 #include<stack>
11 #include<vector>
12 #include<fstream>
13 #include<assert.h>
14 #endif // _C_H_

AVLNode.h

 1 #ifndef _AVLNODE_H_
 2 #define _AVLNODE_H_
 3
 4 typedef int T;
 5
 6 template<typename T>
 7 struct AVLNode{
 8     int height;     //平衡因子
 9     T data;     //数据域
10     AVLNode<T> *lchild, *rchild;    //指针域
11 };
12 #endif // _AVLNODE_H_