看完了第一篇博客，相信大家对于平衡二叉树的插入调整以及删除调整已经有了一定的了解，下面，我们开始介绍代码部分。
首先，再次提一下使用的结构定义
1 typedef char KeyType; //关键字 2 typedef struct MyRcdType //记录 3 { 4 KeyType key; 5 }RcdType,*RcdArr; 6 typedef enum MyBFStatus //为了方便平衡因子的赋值，这里进行枚举 7 { //RH,EH,LH分别表示右子树较高，左右子树等高，左子树较高 8 RH,EH,LH 9 }BFStatus; 10 typedef struct MyBBSTNode //树结点类型定义 11 { 12 RcdType data; //数据成员 13 BFStatus bf; //平衡因子 14 struct MyBBSTNode *lchild,*rchild; //左右分支 15 }BBSTNode,*BBSTree;
结构定义
1. 旋转
旋转是平衡二叉树的基础。所以我们首先介绍。先看具体代码。RRotate的作用就是以*T为根结点的二叉树向右旋转，LRotate就是向左旋转。
参数说明：*T为待旋转子树的根结点。
1 BBSTree RRotate(BBSTree *T) 2 { 3 BBSTree lchild; 4 lchild = (*T)->lchild; 5 (*T)->lchild = lchild->rchild; 6 lchild->rchild = (*T); 7 (*T) = lchild; 8 } 9 void LRotate(BBSTree *T) 10 { 11 BBSTree rchild; 12 rchild = (*T)->rchild; 13 (*T)->rchild = rchild->lchild; 14 rchild->lchild = *T; 15 *T = rchild; 16 }
左右旋转
2. 失衡调整
在前一篇博客我们已经知道，失衡的情况主要包括了两种，左子树过高与右子树过高。我们按其对代码进行划分，首先介绍左子树过高的情况，我们称为左平衡处理。
在这里，不得不再提一下，我们将左失衡，即做左子树过高的情况分为了三种，插入新的结点时，可能出现的情况为LL与LR，删除的时候可能出现的情况为LL,LR,LE。假如不明白为何LE只出现在删除的时候出现，请查看上一篇博客的插入调整与删除调整部分的内容。
函数先行条件：*T为根的树为不平衡子树。
参数说明：*T为不平衡子树的根结点
函数大致说明：首先根据LL,LR,LE对不平衡树进行划分，设定旋转后的最终平衡因子，再对*T进行相应旋转。例如，LL型的不平衡树，最后为*T的左孩子作为根结点，首先设定为最终的值，*T的左孩子的平衡因子为EH，初始根的平衡因子也为EH。
void LeftBalance(BBSTree *T) { BBSTree lchild,rchild; lchild = (*T)->lchild; switch (lchild->bf) { case EH: (*T)->bf = lchild->bf = LH; lchild->bf = RH; RRotate(T); break; case LH: (*T)->bf = lchild->bf = EH; RRotate(T); break; case RH: rchild = lchild->rchild; switch (rchild->bf) { case LH: (*T)->bf = RH; lchild->bf = EH; break; case RH: (*T)->bf = EH; lchild->bf = LH; break; case EH: (*T)->bf = EH; lchild->bf = EH; } rchild->bf = EH; LRotate(&((*T)->lchild)); RRotate(T); break; } }
左平衡调整
1 void RightBalance(BBSTree *T) 2 { 3 BBSTree rchild,lchild; 4 rchild = (*T)->rchild; 5 switch (rchild->bf) 6 { 7 case RH: 8 (*T)->bf = rchild->bf = EH; 9 LRotate(T); 10 break; 11 case EH: 12 (*T)->bf = RH; 13 rchild->bf = LH; 14 LRotate(T); 15 break; 16 case LH: 17 lchild = rchild->lchild; 18 switch (lchild->bf) 19 { 20 case LH: 21 rchild->bf = RH; (*T)->bf = EH; 22 break; 23 case RH: 24 rchild->bf = EH; (*T)->bf = LH; 25 break; 26 case EH: 27 rchild->bf = EH; (*T)->bf = EH; 28 break; 29 } 30 lchild->bf = EH; 31 RRotate(&((*T)->rchild)); 32 LRotate(T); 33 break; 34 } 35 }
右平衡调整
3. 插入新的结点
说明：在插入新的结点的时候，我们使用一个taller的变量来记录树的高度是否变化。默认认为树的高度是有增加的。我们在插入新的结点后，首先判断树的高度是否增加了，假如树的高度没有变化，不必进行如何操作。当树的高度增加时，我们就考虑是否需要对树的进行平衡调整。假如原本根的平衡因子为LH，而插入点又在左子树上，并且子树的高度变高了的时候，我们就要进行左平衡处理。相对的，假如原本根的平衡因子为RH，而插入点又在右子树上，并且子树的高度变高了的时候，我们就要进行右平衡处理。而我们又知道，只需要对最小失衡树进行平衡调整，所以调整后要将taller置为FALSE
参数说明：*T，待插入的平衡二叉树
e，带插入的新的结点的值
taller，*T的子树的高度是否变高的标志。
Status InsertAVL(BBSTree *T,RcdType e,Status *taller) { if(!(*T)) //新建一个节点 return CreatBBSTNode(T,e); else if(e.key == (*T)->data.key) { *taller = FALSE; return TRUE; } if(e.key < (*T)->data.key) //插入到左子树 { Status sign = InsertAVL(&(*T)->lchild,e,taller); if(FALSE == sign || OVERFLOW == sign) return FALSE; if(TRUE == *taller) { switch ((*T)->bf) { case LH: LeftBalance(T); *taller = FALSE; break; case EH: (*T)->bf = LH; *taller = TRUE; break; case RH: (*T)->bf = EH; *taller = FALSE; break; } } } else //插入到了右子树 { Status sign = InsertAVL(&(*T)->rchild,e,taller); if(FALSE == sign || OVERFLOW == sign) return FALSE; if(TRUE == *taller) { switch ((*T)->bf) { case LH: (*T)->bf = EH; *taller = FALSE; break; case EH: (*T)->bf = RH; *taller = TRUE; break; case RH: RightBalance(T); *taller = FALSE; break; } } } return TRUE; }
插入新的结点
4. 删除
说明：平衡二叉树也是一棵二叉查找树，所以其删除操作与二叉查找树是一致的。只是我们需要进行平衡处理。我们使用bfChild记录待删除结点的的子树的根结点的原平衡因子。新的平衡因子由子树的根结点的bf成员进行记录。当待删除的结点处于当前节点的左分支上时，删除结点后，我们调用DelLeftCase设定树的平衡因子以及对树进行调整。当待删除的结点处于当前节点的右分支上时，删除结点后，我们调用DelRightCase设定树的平衡因子以及对树进行调整。而我们需要对结点的平衡因子进行重新设定，只有在子树的高度有所降低时进行。而子树的高度降低，对应着Del***Case函数中的子树是否变为NULL或者子树的平衡因子从LH或者RH变为EH。
参数说明：*T为待进行调整的子树的根结点
bfChild为*T的左孩子在删除结点前的平衡因子
1 //参数说明：*T为待进行调整的子树的根结点 2 //bfChild为*T的右孩子在删除结点前的平衡因子 3 void DelLeftCase(BBSTree *T,int bfChild) 4 { 5 //当bf为-1或1变为0，或者孩子为空时说明子树高降低 6 if((!(*T)->lchild) || (EH != bfChild && EH == (*T)->lchild->bf)) 7 { 8 switch ((*T)->bf)//左子树树高降低 9 { 10 case EH: 11 (*T)->bf = RH; 12 break; 13 case LH: 14 (*T)->bf = EH; 15 break; 16 case RH: //原本右子树比较高 17 RightBalance(T); 18 break; 19 } 20 } 21 } 22 23 void DelRightCase(BBSTree *T,int bfChild) 24 { 25 //当bf为LH或RH变为EH，或者孩子为空时说明子树高降低 26 if((!(*T)->rchild) || (EH != bfChild && EH == (*T)->rchild->bf)) 27 { 28 switch ((*T)->bf) 29 { 30 case EH: 31 (*T)->bf = LH; 32 break; 33 case RH: 34 (*T)->bf = EH; 35 break; 36 case LH: //原本左子树比较高 37 LeftBalance(T); 38 break; 39 } 40 } 41 } 42 BBSTree DeleteNode(BBSTree *T,KeyType key) 43 { 44 int bfChild; 45 if(*T) 46 { 47 if((*T)->data.key > key) 48 { 49 bfChild = (*T)->lchild->bf; 50 (*T)->lchild = DeleteNode(&(*T)->lchild,key); 51 DelLeftCase(T,bfChild); 52 } 53 else if((*T)->data.key < key) 54 { 55 bfChild = (*T)->rchild->bf; 56 (*T)->rchild = DeleteNode(&(*T)->rchild,key); 57 DelRightCase(T,bfChild); 58 } 59 else//当前节点就是要删除的节点 60 { 61 if((*T)->lchild) //*T不是叶子结点并且具有直接前驱 62 { 63 BBSTree farRight = GofarRight((*T)->lchild); 64 (*T)->data = farRight->data; 65 //可以确定，删除的节点为当前节点的左子树的某一个节点 66 (*T)->lchild = DeleteNode(&(*T)->lchild,farRight->data.key); 67 DelLeftCase(T,bfChild); 68 } 69 else if((*T)->rchild) //*T不是叶子结点并且具有直接后驱 70 { 71 BBSTree farLeft = GofarLeft((*T)->rchild); 72 (*T)->data = farLeft->data; 73 (*T)->rchild = DeleteNode(&(*T)->rchild,farLeft->data.key); 74 DelRightCase(T,bfChild); 75 } 76 else //*T是叶子结点 77 { 78 free(*T); 79 *T = NULL; 80 } 81 } 82 } 83 return (*T);//包含了返回NULL与正常的当前节点 84 }
删除
假如您看了图解篇以及代码篇，或许对AVL有了一定的了解。不过还是建议将所有的代码自己实现一次。下面附上完整测试代码的下载链接：http://yunpan.cn/cwUVwIYbPwsC8 访问密码 4302。
测试代码大概解释：使用#表示空结点。代码先自动生成一棵平衡二叉树，首先删除一个结点，用户输入Y就会再次删除平衡二叉树的一个结点。每次删除完了会将树进行输出。输入的格式为A【B,C】的格式，A代表根结点，B为A的左孩子，C为A的右孩子。输入其他字符可以查看最终结果。
PS:假如代码无法编译，请检查随机函数在你的编译器是否可用。
欢迎各位朋友批评改正，谢谢。
