平衡二叉树（AVL树）深入解读

最新推荐文章于 2024-06-08 22:34:04 发布

强迫症Kobe

最新推荐文章于 2024-06-08 22:34:04 发布

阅读量1.1w

点赞数 11

分类专栏： java基础文章标签：平衡二叉树

java基础专栏收录该内容

50 篇文章 1 订阅

订阅专栏

平衡二叉树又称AVL树

性质：

它或者是颗空树，或者是具有下列性质的二叉树：

它的左子树和右子树都是平衡二叉树，且左子树和右子树的深度之差的绝对值不超过1。
若将二叉树节点的平衡因子BF定义为该节点的左子树的深度减去它的右子树的深度，则平衡二叉树上所有节点的平衡因子只可能为-1,0,1.
只要二叉树上有一个节点的平衡因子的绝对值大于1，那么这颗平衡二叉树就失去了平衡。

这里写图片描述
根据上述性质我们可以发现图（a）是一棵平衡二叉树，而图（b）是一棵不平衡二叉树。图中结点的数值代表的就是当前结点的平衡因子。也验证了上述性质，一棵平衡二叉树的所有结点的平衡因子只可能是-1、0、1三种。

为什么需要平衡二叉树？

当然，我们都希望所有的二叉排序树的初始序列都是平衡的，因为平衡二叉树上的任何一个结点左右字数的深度之差都不会超过1，则可以证明它的深度和logn是同数量级的，所以其平均查找长度也和logn同数量级。但是事于愿违有些二叉排序树的插入,或者初始序列由于其插入的先后顺序等缘故，将导致我们的二叉排序树的效率大大降低。如下图
这里写图片描述
为了避免这种情况的发生，我们希望可以有一种算法，将我们的不平衡的二叉排序树转化为平衡二叉排序树。这样就可以让我们的二叉排序树结构最优化。

平衡二叉树的算法

看到上述例子，我们就慢慢有点感觉了，至少知道了为什么会需要平衡二叉树，接下来我们看一下平衡二叉树是怎样将我们不平衡的树转换为平衡二叉树。
如何时构成的二叉排序树编程平衡二叉树呢？先看一个具体的例子
这里写图片描述

图a 一颗空树也算是平衡二叉树
图b 只有一个结点13的树也算是平衡二叉树
图c 在图b的基础上插入新的结点24之后，仍然是平衡二叉树，只是根结点的平衡因子从0变到了-1（左子树的深度为0减去右子树的深度1等于-1）
图d 在图c的基础上再插入一个结点37，这个时候整棵树出现了不平衡现象，根结点13的平衡因子从-1变成了-2。我们想要让这课树平衡，而且要保证该树二叉排序树的性质，那么我们只要将根结点13换为24，结点13作为结点24的左子树，这棵树就又会回到平衡状态，如图e。我们把这种对树做向左逆时针“旋转”的操作称为单向左旋平衡处理。左旋之后，我们发现13、24、37结点的平衡因子都变为0。而且仍然保持着二叉排序树的特性
图f当我们继续插入结点90之后，二叉树仍然平衡，只是24、37两个结点的平衡因子变为了-1，再次插入53结点之后，结点37的平衡因子BF由-1变为-2，这意味着该排序树中出现了新的不平衡现象，需要进行调整。但此时由于结点53插在结点90的左子树上，因此不能如上面一样作简单的调整。对于以上结点37为根的子树来说，既要保持二叉排序树的特性，又要平衡，则必须以53结点作为根结点，而使3**7结点成为它左子树的根，**90结点称为它右子树的根。这就好比做了两次旋转，首先我们让37、53、90这棵树单先向右顺时针转变成图g，再像左逆时针变成图h，这样我们的二叉树就能够再次回到平衡状态。对于以上旋转操作我们称为双向旋转（先右后左）平衡处理

平衡算法总结

看完了上面的例子，我们总结一下二叉排序树的不平衡情况以及如何将其转化为平衡情况。
一般情况下，假设由于在二叉排序树上插入结点而失去平衡的最小子树根结点的指针为a（即a是离插入结点最近，且平衡因子绝对值不超过1的祖先结点），则失去平衡后进行调整的规律可以归纳为一下4种情况：

单向右旋平衡处理：由于在a的左子树根结点的左子树上插入结点，a的平衡因子由1增加到2，致使以a为根结点的子树失去平衡，则需要进行一次右向顺时针旋转操作。简称LL型旋转
单想左旋平衡处理：由于在a的右子树根结点的右子树上插入结点，a的平衡因子由-1增加到-2，致使以a为根结点的子树失去平衡，则需要进行一次左向逆时针旋转操作。简称RR型旋转
双向旋转（先左后右）平衡处理：由于在a的左子树的根结点的右子树上插入结点，a的平衡因子由1增加到2，致使a为根结点的子树失去平衡，则需要进行两次旋转（先左旋后右旋）操作。简称LR型旋转
双向旋转（先右后左）平衡处理：由于在a的右子树的根结点的左子树上插入结点，a的平衡因子由1增加到2，致使a为根结点的子树失去平衡，则需要进行两次旋转（先右旋后左旋）操作。简称RL型旋转

如何证明我们插入的正确性：中序遍历所得关键字的值序列从小到大即可（二叉排序树的性质）

那么如何创建一颗平衡二叉树呢？

创建平衡二叉树，我们采用依次插入节点的方式进行。而平衡二叉树上插入节点采用递归的方式进行。递归算法如下：

（1）若该树为一空树，那么插入一个数据元素为e的新节点作为平衡二叉树的根节点，树的高度增加1。

（2）若待插入的数据元素e和平衡二叉树（BBST）的根节点的关键字相等，那么就不需要进行插入操作。

（3） 若待插入的元素e比平衡二叉树（BBST）的根节点的关键字小，而且在BBST的左子树中也不存在和e有相同关键字的节点，则将e插入在BBST的左子树上，并且当插入之后的左子树深度增加1时，分别就下列情况处理之。

BBST的根节点的平衡因子为-1（右子树的深度大于左子树的深度）：则将根节点的平衡因子更改为0（左旋），BBST的深度不变；
BBST的根节点的平衡因子为0（左右子树的深度相等）：则将根节点的平衡因子修改为1（不用旋），BBST的深度增加1；
BBST的根节点的平衡因子为1（左子树的深度大于右子树的深度）：若BBST的左子树根节点的平衡因子为1，则需要进行单向右旋转平衡处理，并且在右旋处理后，将根节点和其右子树根节点的平衡因子更改为0，树的深度不变；
若BBST的左子树根节点的平衡因子为-1，则需进行先向左，后向右的双向旋转平衡处理，并且在旋转处理之后，修改根节点和其左，右子树根节点的平衡因子，树的深度不变；

（4）若e的关键字大于BBST的根节点的关键字，而且在BBST的右子树中不存在和e有相同关键字的节点，则将e插入到BBST的右子树上，并且当插入之后的右子树深度加1时，分别就不同的情况处理之。

BBST的根节点的平衡因子是1（左子树的深度大于右子树的深度）：则将根节点的平衡因子修改为0（右旋），BBST的深度不变；
BBST的根节点的平衡因子是0（左右子树的深度相等）：则将根节点的平衡因子修改为-1（不用旋），树的深度加1；
BBST的根节点的平衡因子为-1（右子树的深度大于左子树的深度）：若BBST的右子树根节点的平衡因子为1，则需要进行两次选择，第一次先向右旋转，再向左旋转处理，并且在旋转处理之后，修改根节点和其左，右子树根节点的平衡因子，树的深度不变；
若BBST的右子树根节点的平衡因子为1，则需要进行一次单向左的旋转处理，并且在左旋之后，更新根节点和其左，右子树根节点的平衡因子，树的深度不变；

代码实现

/*********************************************************
-  Copyright (C): 2016
-  File name    : bbst.c
-  Author       : - zxn - 
-  Date         : 2016年08月06日 星期六 15时45分15秒
-  Description  : 平衡二叉树                  
*  *******************************************************/
#include <stdio.h>
#include <stdlib.h>
#include <stdbool.h>

#define LH +1   //左高
#define EH 0    //等高
#define RH -1   //右高

//二叉树类型定义
struct tree
{
    int data;             //数据域
    int bf;               //平衡因子
    struct tree *left;    //左孩子
    struct tree *right;   //右孩子
};

typedef struct tree treenode;
typedef treenode *btree;

//右旋
//对以*ptr为根的二叉排序树做右旋处理，处理之后p指向新的树根结点，即旋转
//处理之前左子树的根结点
void R_Rotate(btree *ptr)
{
    btree lc = (*ptr)->left;   //lc指向的*ptr的左孩子的根结点
    (*ptr)->left = lc->right;  //lc的右子树挂接为*ptr的左子树
    lc->right = *ptr;          
    *ptr = lc;                 //ptr指向新的结点"
}

//左旋
//对以*ptr为根的二叉排序树做左旋处理，处理之后p指向新的树根结点，即旋转
//处理之前右子树的根结点
void L_Rotate(btree *ptr)
{
    btree rc = (*ptr)->right;   //rc指向的*ptr的由孩子的根结点
    (*ptr)->right = rc->left;   //rc的左子树挂接为*ptr的右子树
    rc->left = *ptr;
    *ptr = rc;                  //ptr指向新的结点
}

/*
以指针root所指结点为根结点的二叉树作左平衡旋转处理，本算法结束时，指针root指向新的结点
*/
void LeftBalance(btree *root)
{

    btree lc;
    btree rd;

    lc = (*root)->left; //ls指向*root的左根结点

    //检测*root的左子树的平衡度，并作相应处理
    switch (lc->bf) 
    {
        case LH:
        {
            //新结点插入在*root的左孩子的左子树上，要做单右旋处理
            (*root)->bf = lc->bf = EH;
            R_Rotate(root);
            break;
        }
        case RH:
        {
            //新结点插入在*root左孩子的右子树上要做双旋处理
            //rd指向*t的左孩子的右子树根上
            rd = lc->right;
            switch(rd->bf)
            {
                //修改*root及其左孩子的平衡因子
                case LH:
                {
                    (*root)->bf = RH;
                    lc->bf = EH;
                    break;
                }
                case EH:
                {
                    (*root)->bf = lc->bf = EH;
                    break;
                }
                case RH:
                {
                    (*root)->bf = EH;
                    lc->bf = LH;
                    break;
                }
            }
            rd->bf = EH;
            //对*root的左子树左左旋平衡处理
            L_Rotate(&(*root)->left);
            //对*root做右旋平衡处理
            R_Rotate(root);
            break;
        }
    }
}

void RightBalance(btree *root)
{
    btree lc;
    btree rd;

    lc = (*root)->right;

    switch (lc->bf)
    {
        case RH:
        {
            (*root)->bf = lc->bf = EH;
            L_Rotate(root);
            break;
        }
        case LH:
        {
            rd = lc->left;
            switch(rd->bf)
            {
                case LH:
                {
                    (*root)->bf = EH;
                    lc->bf = RH;
                    break;
                }
                case EH:
                {
                    (*root)->bf = lc->bf = EH;
                    break;
                }
                case RH:
                {
                    (*root)->bf = LH;
                    lc->bf = EH;
                    break;
                }
            }
            rd->bf = EH;
            R_Rotate(&(*root)->right);
            L_Rotate(root);
            break;
        }
    }
}

/* 
 平衡二叉排序树的插入
 若在平衡的二叉树root中不存在和e相同关键字的结点，则插入一个数据元素为e的新结点，并返回1，否则返回0，若因插入而使二叉树失去平衡，则作平衡处理，taller变量反应T长高与否
 若树长高，则置taller为ture
 */
int InsertAVL(btree *root, int e, bool *taller)
{
    if ((*root) == NULL)
    {
        //该树为一棵空树，创建一个新节点作为根节点
        (*root) = (btree)malloc(sizeof(treenode));
        (*root)->bf = EH;
        (*root)->data = e;
        (*root)->left = NULL;
        (*root)->right = NULL;
        *taller = true;
    }
    else if (e == (*root)->data)
    {
        //关键字相同，则不再继续插入
        *taller = false;
        return 0;
    } 
    else if (e < (*root)->data)
    {
        //应该继续在*root的左子树进行搜索
        if (!InsertAVL(&(*root)->left, e, taller))
        {
            //未插入
            return 0;
        }
        //已插入到*root的左子树中并且左子树长高
        if (*taller)
        {
            //检查*root的平衡度
            switch ((*root)->bf)
            {
                //原本左子树比右子树高
                case LH:
                {
                    //平衡因子为-1
                    //左旋
                    LeftBalance(root);
                    *taller = false;
                    break;
                }
                //原本左右树一样高，现在因为左子树长高树长高
                case EH:
                {
                    //平衡因子为0
                    (*root)->bf = LH;
                    *taller = true;
                    break;
                }
                //原本右子树比左子树高，现在等高
                case RH:
                {
                    //平衡因子为1
                    (*root)->bf = EH;
                    *taller = false;
                    break;
                }
            }
        }
    }
    else
    {
        //应继续在*root的右子树中进行搜索    
        if (!InsertAVL(&(*root)->right, e, taller))
        {
            //未插入
            return 0;
        }
        //已插入到*root的右子树且右子树长高
        if (*taller)
        {
            //检查*root的平衡度
            switch((*root)->bf)
            {
                case LH:
                {
                    //原本左子树比右子树高，现在相等
                    (*root)->bf = EH;
                    *taller = false;
                    break;
                }
                case EH:
                {
                    //原来左右子树登高，现在因为右子树长高树长高
                    (*root)->bf = RH;
                    *taller = true;
                    break;
                }
                case RH:
                {
                   //原本右子树比左子树高，需要做右旋平衡处理 
                    RightBalance(root);
                    *taller = false;
                    break;
                }
            }
        }
    }
    return 1;
}

/*
二叉树的中序遍历
*/
void inorder(btree root)
{
    if (root->left)
    {
        inorder(root->left);
    }
    printf("%d    ", root->data);
    if (root->right)
    {
        inorder(root->right);
    }
}

/*
 平衡二叉树的查找
 */

bool FindNode(btree root, int value, btree *pos)
{
    btree ptr = root;

    (*pos) = NULL;

    while (ptr)
    {
        if (ptr->data == value)
        {
            //找到了
            (*pos) = ptr;
            return true;
        }
        else if (ptr->data > value)
        {
            ptr = ptr->left;
        }
        else
        {
            ptr = ptr->right;
        }
    }
    //没有找到
    return false;
}

int main()
{
   int i;
   int value;
   int nArr[] = {1,23,45,34,98,9,4,35,23};

   bool taller;

   btree root = NULL;
   btree pos = NULL;

   for (i = 0; i < 9; i++)
   {
       //插入数据
       InsertAVL(&root, nArr[i], &taller);
   }
   printf("---------中序遍历结果---------\n");
   inorder(root);

   printf("\n请输入查找的数：");
   scanf("%d", &value);

   if (FindNode(root, value, &pos))
   {
       printf("\n找到了:%d\n", pos->data);
   }
   else
   {
       printf("\nNot find this node\n");
   }

   return 0;
}

插入过程
这里写图片描述

代码结果
这里写图片描述

这里写图片描述

强迫症Kobe

关注

11
点赞
踩
35

收藏

觉得还不错? 一键收藏
4
评论
平衡二叉树（AVL树）深入解读

平衡二叉树又称AVL树性质：它或者是颗空树，或者是具有下列性质的二叉树：它的左子树和右子树都是平衡二叉树，且左子树和右子树的深度之差的绝对值不超过1。若将二叉树节点的平衡因子BF定义为该节点的左子树的深度减去它的右子树的深度，则平衡二叉树上所有节点的平衡因子只可能为-1,0,1. 只要二叉树上有一个节点的平衡因子的绝对值大于1，那么这颗平衡二叉树就失去了平衡。根据上述性质我...
复制链接

扫一扫