平衡二叉树（AVL树）

平衡二叉树是一种二叉排序树，其中每一个节点的左子树和右子树的高度至多等于1。

平衡因子：将二叉树上结点的左子树深度减去右子树深度的值称为平衡因子（Balance Factor）。

距离插入结点最近的，且平衡因子的绝对值大于1的结点为根的子树，我们称为最小不平衡子树。如下图所示：

1.平衡二叉树实现原理

平衡二叉树构建的基本思想就是在构建二叉排序树的过程中，每当插入一个结点时，先检查是否因插入破坏了树的平衡性。若是，则找出最小不平衡子树。在保持二叉排序树特性的前提下，调整最小不平衡子树中各结点之间的链接关系，进行相应的旋转，使之成为新的平衡子树。

2.平衡二叉树实现算法

首先改进二叉排序树的结点结构，增加一个bf，用来存储平衡因子。

/* 二叉树的二叉链表结构定义 */
typedef struct BiTNode   //结点结构
{
    int data;
    int bf;       //结点平衡因子
    struct BiTNode *lchild, *rchild; //左右孩子指针
} BiTNode, *BiTree;

对于右旋操作，代码如下：

/* 对以p为根的二叉排序树作右旋处理 */
/* 处理之后p指向新的树根结点，即旋转处理之前的左子树的根结点 */
void R_Rotate(BiTree *P)
{
    BiTree L;
    L = (*P)->lchild;  //L指向P的左子树根结点
    (*P)->lchild = L->rchild; //L的右子树挂接为P的左子树
    L->lchild = (*P);
    *P = L;               //P指向新的根结点
}

左旋代码如下：

/* 对以p为根的二叉排序树作左旋处理 */
void L_Rotate(BiTree *P)
{
    BiTree R;
    R = (*P)->rchild; // R指向P的右子树根结点
    (*P)->rchild = R->lchild;  // R的左子树挂接为P的右子树
    R->rchild = (*P)；
    *P = R;
}

左平衡旋转的代码如下：

#define LH +1 //左高
#define EH 0  //等高
#define RH -1 //右高
/* 对以指针T所指结点为根的二叉树作左平衡旋转处理 */
/* 本算法结束时，指针T指向新的根结点 */
void LeftBalance(BiTree *T)
{
    BiTree L,Lr;
    L = (*T)->lchild; //L指向T的左子树根结点
    switch(L->br)
    { /* 检查T的左子树平衡度，并作相应平衡处理*/
        case LH: //新结点插入在T的左孩子树的左孩子树上，作单右旋处理
            (*T)->bf = L->bf = EH;
            R_Rotate(T);
            break;
        case RH: //新结点插入在T的左孩子的右孩子树上，要作双旋处理
            Lr = L->rchild; //Lr指向T的左孩子的右子树根
            switch(Lr->bf)
            {
                case LH: (*T)->bf = RH;
                        L->bf = EH;
                            break;
                case EH: (*T)->bf = L->bf = EH;
                        break;
                case RH: (*T)->bf = EH;
                        L->bf = LH;
                        break;
            }
            Lr->bf = EH;
            L-Rotate(&(*T)->lchild);  //对T的左子树作左旋平衡处理
            R_Rotate(T);   //对T作右旋平衡处理
    }
}

同样地，右旋平衡处理的函数代码与上面类似。有了上面这些，主函数呼之欲出。

/* 若在平衡二叉树T中不存在和e有相同关键字的结点，则插入一个数据元素为
   1的新结点并返回1，否则返回0,。若因插入而使二叉排序树失去平衡，则作
   平衡旋转处理，布尔变量taller反映T长高与否
*/
Status InsertAVL(BiTree *T, int e, Status *taller)
{
    if(!*T)
    {  //插入新结点，树“长高”，置taller为TRUE
        *T = (BiTree)malloc(sizeof(BiTNode));
        (*T)->data = e;
        (*T)->lchild = (*T)->rchild = NULL:
        (*T)->bf = EH;
        *taller = TRUE;    
    }
    else
    {
        if (e==(*T)->data)
        { //树中已存在和e有相同关键字的结点则不再插入
            *taller = FALSE;
            return FALSE;        
        }
        if (e<(*T)->data)
        {  //继续在T的左子树中进行搜索
            if (!InsertAVL(&(*T)->lchild, e, taller))  //未插入
                return FALSE;
            if (taller) //已插入到T的左子树中且左子树“长高”
            {
                switch((*T)->bf)  //检查T的平衡度
                {
                    case LH:  //原本左子树比右子树高，需要作左平衡处理
                            LeftBalance(T);
                            *taller = FALSE;
                            break;
                    case EH:  //原本左右子树等高，现因左子树增高而树增高
                            (*T)->bf = LH;
                            *taller = TRUE;
                            break;
                    case RH: //原本右子树比左子树高，现在左右子树等高
                            (*T)->bf = EH;
                            *taller = FALSE;
                            break;
        
                }
            }
            
        }
        else
        { //应继续在T的右子树中进行搜索
            if (!InsertAVL(&(*T)->lchild, e, taller))   //未插入
                return FALSE;
            if (*taller)
            {
                switch((*T)->bf) //检查T的平衡度
                {
                    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;
}
   

对于这段代码来说，我们只需要执行如下代码即可在内存中生成一颗平衡二叉树。

int i;
int a[10] = {3,2,1,4,5,6,7,10,9,8}
BiTree = NULL;
Status taller;
for (i=0;i<10;i++)
{
    InsertAVL(&T, a[i], &taller);
}