当前位置:   article > 正文

数据结构—平衡二叉树(AVL树)的原理以及Java代码的完全实现_平衡二叉树原理是什么

平衡二叉树原理是什么

平衡二叉树(AVL树),顾名思义,是一颗很“平衡”的树,它的平衡是相对于排序二叉树来说的。为了避免极端情况下二叉搜索树节点分布不均匀,甚至退化为链表,影响查找效率,我们引入了平衡二叉树,即让树的结构看起来尽量“均匀”,左右子树的节点数和层级尽量一样多。

本文详细介绍了平衡二叉树的概念和实现原理,并且提供了Java代码的完全实现。

1 平衡二叉树的概述

为了避免极端情况下二叉搜索树退化为链表,影响查找效率,我们引入了平衡二叉树,即让树的结构看起来尽量“均匀”,左右子树的节点数和层级尽量一样多。要想学习平衡二叉树并且掌握它,必须要先掌握二叉排序树,如果对二叉搜索树还不太明白的,包括为什么二叉排序树可能退化为链表,可以看看这篇文章:数据结构—二叉排序树的原理以及Java代码的完全实现

平衡二叉树,又称AVL树,指的是左子树上的所有节点的值都比根节点的值小,而右子树上的所有节点的值都比根节点的值大,且左子树与右子树的高度差最大为1。因此,平衡二叉树满足所有二叉排序(搜索)树的性质,是在二叉排序树的基础上发展而来的。至于AVL,则是取自两个发明平衡二叉树的俄罗斯科学家的名字:G. M. Adelson-Velsky和E. M. Landis。

总的来说平衡二叉树具有如下性质:

  1. 它一定是一棵二叉排序树
  2. 它是一棵空树或它的左右两个子树的高度差的绝对值不超过1,并且左右两个子树都是一棵平衡二叉树,递归定义。

平衡因子BF(Balance Factor): 我们将二叉树上节点的左子树深度减去右子树深度的值称为平衡因子,那么平衡二叉树上所有节点的平衡因子只可能是-1、0和1。只要二叉树上有一个节点的平衡因子的绝对值大于1,则该二叉树就是不平衡的。

在这里插入图片描述

上图中,图一是平衡二叉树,图二的59比58大,却是58的左子树,这是不符合二叉排序树的定义的,图二不是平衡二叉树。图3不是平衡二叉树的原因就在于,节点58的左子树高度为3,而右子树为空,二者差大于了绝对值1,因此它也不是平衡的。而经过适当的调整后的图4,它就符合了定义,因此它是平衡二叉树。

最小不平衡子树: 距离插入、删除节点最近的,且平衡因子的绝对值大于1的节点为根的子树,我们称为最小不平衡子树。下图中当新插入节点37时,距离它最近的平衡因子绝对值超过1的节点是58(即它的左子树高度3减去右子树高度1),所以从58开始以下的子树为最小不平衡子树。

在这里插入图片描述

2 平衡二叉树的实现原理

平衡二叉树实现原理的核心就是:由于在插入、删除节点以后,只有那些从插入点到根节点的路径上的节点的平衡可能被改变,因为只有这些节点的子树可能发生变化。因此,我们需要沿着这条路径上行到根并更新平衡信息,尝试找出最小不平衡树。在保持二叉排序树特性的前提下,调整最小不平衡子树中根节点和子结点之间的关系,进行相应的旋转(rotation),使之成为新的平衡子树。

先来看看插入的重平衡,因为到后面我们会发现插入和删除进行的重平衡操作基本是一致的。

我们把需要进行平衡(平衡因子绝对值大于1)的节点称为x,由于任意节点最多有两个儿子,因此出现高度不平衡就需要x点的两棵子树的高度差2,而这种不平衡只可能出现在下面四种情况中:

  1. 在节点X的左孩子节点的左子树中插入元素,简称LL
  2. 在节点X的左孩子节点的右子树中插入元素,简称LR
  3. 在节点X的右孩子节点的左子树中插入元素,简称RL
  4. 在节点X的右孩子节点的右子树中插入元素,简称RR

其中第1种情况和第4种情况是对称的,被称为发生“外边”的情况,可以通过单旋转来解决,而第2种情况和第3种情况是对称的,被称为发生在“内边”的情况,需要双旋转来解决。

案例:对数组中的元素{3,2,1,4,5,6,7,16,15,14,13,12,11,10,8,9}顺序插入并建立一个平衡二叉树。以这个案例为例,来讲解上面4个问题的通用解决办法和单旋转和双旋转的概念。

2.1 单旋转

首先是添加前两个元素“3、2”的时候,可以正常的构建平衡二叉树,到了第3个数“1”时,发现此时根节点“3”的平衡因子变成了2,此时整棵树都成了最小不平衡子树,因此需要调整结构。

在这里插入图片描述

上图中的情况情况符合条件1——LL,因此所采用单旋转来重平衡。此时,我们需要右旋(顺时针旋转)。旋转的目的实际上就是为了降低深度,保持平衡。

在这里插入图片描述

节点3经过右旋后,节点2变成了根节点,节点3变成了2的右子树,此时树节点1的深度降低了一级,整颗树重新回到了平衡。我们把通过一次旋转即可修复平衡的操作叫做单旋转。

平衡因子BF绝对值大于1的节点X称为失衡点,修复一棵被破坏的AVL树时,找到失衡点是很重要的,查找失衡点就是从新插入、删除的节点的位置向上回溯至根节点的过程。

然后我们再增加节点4,平衡因子没有超出限定范围。增加节点5时,节点3的BF值为-2,说明又要旋转了。

在这里插入图片描述

上图中的情况情况符合条件4——RR,需要采用单旋转来重平衡。此时,我们需要左旋(逆时针旋转)。

在这里插入图片描述

左旋之后,如上图右,树的深度降低了一级,此时整棵树又达到了平衡状态。继续,增加节点6时,发现根节点2的BF值变成了-2,所以我们对根节点进行了左旋。

在这里插入图片描述

左旋的结果使得节点2成为节点4的左孩子,原本处于2和4之间的节点3是4的左子树,由于旋转后需要满足二叉排序树特性,因此它成了节点2的右子树,因为该子树的每一个关键字都在2-4之间,因此这个变换是成立的。

现在我们来尝试总结出发生情况1和4时的通用解法

首先,情况1和4可以提炼出一个通用模型:

在这里插入图片描述

模型中,左边如果要发生不平衡的情况1,那么左子树1的深度肯定比右子树1的深度2层;右边如果要发生不平衡的情况4,那么左子树1的深度肯定比右子树1的深度2层。

针对上面情况1和情况4,我们分别使用右旋和左旋,来降低或者升高这两颗子树的深度:

在这里插入图片描述

如上图,情况1右旋之后,k2成为根节点,k1成为k2的右子节点,k2的右子树2成为k1的左子树;情况4左旋之后,k2成为根节点,k1成为k2的左子节点,k2的左子树2成为k1的右子树。树重新达到了平衡状态,这就是解决情况1和情况4的通解,并且我们可以发现它们是对称的。

下面增加节点7,这导致节点5的BF变成了-2,且符合情况4,需要左旋,根据上面的通解,采用下面的左旋方法让树重新成为平衡二叉树:

在这里插入图片描述

2.2 双旋转

上面的单旋转对于情况2和3是没有用的,因为此时树结构太深,单旋转并不会减低它的深度。此时需要使用双旋转。

当增加节点16时,结构无变化,再增加节点15,此时节点7的BF变成了-2。此时符合情况3:在节点X的右孩子节点的左子树中插入元素,简称RL。如下图:

在这里插入图片描述

此时简单的左旋无法解决问题:节点15成了16的右孩子,这是不符合二叉排序树的特性的,此时不能简单的左旋。如下图:

在这里插入图片描述

对于这种情况,我们对于关键节点7、16、15先建立一个更广泛的模型:
在这里插入图片描述
其中7-k1、16-k2、15-k3,并且节点7完全还可以拥有左子树,节点16可以拥有右子树,而节点15则可以拥有左右子树。

要想发生上面k1的BF为-2的情况,需要左子树2或右子树2其中一颗子树的深度比左子树1深两层,或者他们都是空子树,但是我们不知道是具体是什么情况,不过这没关系,在这里我们要求出一个对这个问题通解!

此时为了平衡高度,我们不能将k1当作根节点了,但是左旋——把k2当作根节点也不能解决问题(上面已经证实了),唯一的选择就是:将k3当作新的根节点,并且先使得k2右旋成为k3的右子树,然后k1左旋成为k3的左子树,并且左子树2成为k1的右子树,右子树2成为k2的左子树,这是完全成立的,这就是情况3的通解。 最终,右-左双旋结果如下:

在这里插入图片描述

我们可以看到,无论是具体发生了什么情况(左子树2或右子树2其中一颗子树的深度比左子树1深两层,或者他们都是空子树),左-右双旋转换为上右图的形状之后,左子树2或右子树2都会被削减一层深度,而左子树1会被增加一层深度,这棵树始终都是一颗平衡二叉树。

实际上,右-左双旋,分开旋转的过程模型如下:

在这里插入图片描述

回到案例,案例中左子树2、右子树2、左子树1、右子树1都是空树,使用右-左双旋之后,树结构如下图,该树得以重新平衡:

在这里插入图片描述

接着插入14,情况与刚才类似,节点6的BF是-2,此时符合RL的情况(在节点6的右孩子节点15的左子树7中插入元素),如下图左,此时继续右-左双旋后,整棵树又回到了平衡状态,如下图右:

在这里插入图片描述

继续插入13,此时根节点4的BF变成了-2,符合情况4,此时使用一次单左旋即可解决问题:

在这里插入图片描述

继续插入12之后,向上回溯到节点14时,发现节点14的BF为2,此时符合情况1,需要右旋恢复平衡:

在这里插入图片描述

继续插入11之后,向上回溯到节点15时,发现节点15的BF为2,此时符合情况1,需要右旋恢复平衡:

在这里插入图片描述

继续插入10之后,向上回溯到节点12时,发现节点12的BF为2,此时符合情况1,需要右旋恢复平衡:

在这里插入图片描述

插入8之后,向上回溯到根节点也没有发现最小不平衡树,因此不需要旋转。最后插入9之后,我们发现出现了情况2,此时我们有情况1和情况4对称的经验,自然也知道需要右-左双旋的的对称操作——左-右双旋来重新平衡。

先来看左-右双旋模型:

在这里插入图片描述

它和右-左双旋模型就是对称操作,将k3当作新的根节点,并且先使得k2左旋成为k3的左子树,然后k1右旋成为k3的右子树,并且左子树2成为k2的右子树,右子树2成为k1的左子树,这是完全成立的,这就是情况2的通解。

左-右双旋之后,重新形成了平衡二叉树:

在这里插入图片描述

实际上,左-右双旋,分开旋转的过程模型如下:

在这里插入图片描述

节点添加完毕,最终形成了一颗平衡二叉树:

在这里插入图片描述

2.3 总结

插入节点的不平衡的情况只有四种:

  1. 在节点X的左孩子节点的左子树中插入元素,简称LL
  2. 在节点X的左孩子节点的右子树中插入元素,简称LR
  3. 在节点X的右孩子节点的左子树中插入元素,简称RL
  4. 在节点X的右孩子节点的右子树中插入元素,简称RR

其中1采用单右旋、4采用单左旋即可解决问题。2和3比较复杂,2需要采用左-右双旋、3需要采用右-左双旋。

1和4、2和3是对称的情况,现在综合起来看,所谓的旋转似乎也不那么复杂,并且我们已经求出了这几种问题的通解,该通解对于节点的删除是同样适用的,不必再考虑各种特殊情况,非常方便,下面来看看具体的代码实现!

3 平衡二叉树的构建

3.1 类架构

首先节点对象还是需要一个数据域和两个引用域,相比于二叉排序树,还要多一个节点高度的字段,这样方便计算平衡因子,并且提供返回节点高度的方法。
另外还需要一个比较器的引用,因为需要对元素进行排序,自然需要比较元素的大小,如果外部传递了比较器,那么就使用用户指定的比较器进行比较,否则,数据类型E必须是Comparable接口的子类,否则因为不能比较而报错。

另外,还需要提供中序遍历的方法,该遍历方法对于二叉排序树的结果将会顺序展示。

public class AvlTree<E> {
    /**
     * 外部保存根节点的引用
     */
    private BinaryTreeNode<E> root;

    /**
     * 自定义比较器
     */
    private Comparator<? super E> cmp;


    /**
     * 树节点的数量
     */
    private int size;

    /**
     * 内部节点对象
     *
     * @param <E> 数据类型
     */
    public static class BinaryTreeNode<E> {

        //数据域
        E data;
        //左子节点
        BinaryTreeNode<E> left;
        //右子节点
        BinaryTreeNode<E> right;
        //节点高度 从0开始,从下往上;null节点高度返回-1
        int height;

        public BinaryTreeNode(E data) {
            this.data = data;
        }

        @Override
        public String toString() {
            return data.toString();
        }

    }

    /**
     * 指定比较器
     *
     * @param cmp 比较器
     */
    public AvlTree(Comparator<? super E> cmp) {
        this.cmp = cmp;
    }

    /**
     * 空构造器
     */
    public AvlTree() {
    }

    /**
     * 是否是空树
     *
     * @return true 是 ;false 否
     */
    public boolean isEmpty() {
        return size == 0;
    }

    /**
     * 返回节点数
     *
     * @return 节点数
     */
    public int size() {
        return size;
    }


    /**
     * 对元素进行比较大小的方法,如果传递了自定义比较器,则使用自定义比较器,否则则需要数据类型实现Comparable接口
     *
     * @param e1 被比较的第一个对象
     * @param e2 被比较的第二个对象
     * @return 0 相等 ;小于0 e1 < e2 ;大于0 e1 > e2
     */
    private int compare(E e1, E e2) {
        if (cmp != null) {
            return cmp.compare(e1, e2);
        } else {
            return ((Comparable<E>) e1).compareTo(e2);
        }
    }


    /**
     * 保存遍历出来的节点数据
     */
    List<BinaryTreeNode<E>> str = new ArrayList<>();

    /**
     * 中序遍历,提供给外部使用的api
     *
     * @return 遍历的数据
     */
    public String toInorderTraversalString() {

        //如果是空树,直接返回空
        if (isEmpty()) {
            return null;
        }
        //从根节点开始递归
        inorderTraversal(root);
        //获取遍历结果
        String s = str.toString();
        str.clear();
        return s;
    }

    /**
     * 中序遍历 内部使用的递归遍历方法,借用了栈的结构
     *
     * @param root 节点,从根节点开始
     */
    private void inorderTraversal(BinaryTreeNode<E> root) {

        BinaryTreeNode<E> left = getLeft(root);
        if (left != null) {
            //如果左子节点不为null,则继续递归遍历该左子节点
            inorderTraversal(left);
        }
        //添加数据节点
        str.add(root);
        //获取节点的右子节点
        BinaryTreeNode<E> right = getRight(root);
        if (right != null) {
            //如果右子节点不为null,则继续递归遍历该右子节点
            inorderTraversal(right);
        }
    }

    /**
     * 获取左子节点
     *
     * @param parent 父节点引用
     * @return 左子节点或者null--表示没有左子节点
     */
    public BinaryTreeNode<E> getLeft(BinaryTreeNode<E> parent) {
        return parent == null ? null : parent.left;
    }

    /**
     * 获取右子节点
     *
     * @param parent 父节点引用
     * @return 右子节点或者null--表示没有右子节点
     */
    public BinaryTreeNode<E> getRight(BinaryTreeNode<E> parent) {
        return parent == null ? null : parent.right;
    }

    /**
     * 获取根节点
     *
     * @return 根节点 ;或者null--表示空树
     */
    public BinaryTreeNode<E> getRoot() {
        return root;
    }

    /**
     * 获取height
     *
     * @param node 节点
     * @return 高度或者-1 表示节点为null
     */
    private int getHeight(BinaryTreeNode<E> node) {
        return node == null ? -1 : node.height;
    }

}
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35
  • 36
  • 37
  • 38
  • 39
  • 40
  • 41
  • 42
  • 43
  • 44
  • 45
  • 46
  • 47
  • 48
  • 49
  • 50
  • 51
  • 52
  • 53
  • 54
  • 55
  • 56
  • 57
  • 58
  • 59
  • 60
  • 61
  • 62
  • 63
  • 64
  • 65
  • 66
  • 67
  • 68
  • 69
  • 70
  • 71
  • 72
  • 73
  • 74
  • 75
  • 76
  • 77
  • 78
  • 79
  • 80
  • 81
  • 82
  • 83
  • 84
  • 85
  • 86
  • 87
  • 88
  • 89
  • 90
  • 91
  • 92
  • 93
  • 94
  • 95
  • 96
  • 97
  • 98
  • 99
  • 100
  • 101
  • 102
  • 103
  • 104
  • 105
  • 106
  • 107
  • 108
  • 109
  • 110
  • 111
  • 112
  • 113
  • 114
  • 115
  • 116
  • 117
  • 118
  • 119
  • 120
  • 121
  • 122
  • 123
  • 124
  • 125
  • 126
  • 127
  • 128
  • 129
  • 130
  • 131
  • 132
  • 133
  • 134
  • 135
  • 136
  • 137
  • 138
  • 139
  • 140
  • 141
  • 142
  • 143
  • 144
  • 145
  • 146
  • 147
  • 148
  • 149
  • 150
  • 151
  • 152
  • 153
  • 154
  • 155
  • 156
  • 157
  • 158
  • 159
  • 160
  • 161
  • 162
  • 163
  • 164
  • 165
  • 166
  • 167
  • 168
  • 169
  • 170
  • 171
  • 172
  • 173
  • 174
  • 175
  • 176
  • 177
  • 178
  • 179
  • 180

3.2 查找的方法

平衡二叉树就是一颗二叉排序树,其查找方法可以复用二叉排序树的查找方法,很简单:

  1. 若根节点的关键字值等于查找的关键字,成功,返回true;
  2. 否则,若小于根节点的关键字值,递归查左子树;
  3. 若大于根节点的关键字值,递归查右子树;
  4. 最终查找到叶子节点还是没有数据,那么查找失败,则返回false
    /**
     * 查找,开放给外部使用的api
     * @param e 要查找的元素
     * @return false 不存在 true 存在
     */
    public boolean contains(E e) {
        return contains(e, root);
    }

    /**
     * 查找,内部调用的方法,从根节点开始查找
     *
     * @param e    要查找的元素
     * @param root 节点
     * @return false 不存在 true 存在
     */
    private boolean contains(E e, BinaryTreeNode<E> root) {
        /*null校验*/
        if (root == null) {
            return false;
        }
        /*调用比较的方法*/
        int i = compare(e, root.data);
        /*如果大于0,则说明e>root.date 继续查询右子树*/
        if (i > 0) {
            return contains(e, root.right);
            /*如果小于0,则说明e<root.date 继续查询左子树*/
        } else if (i < 0) {
            return contains(e, root.left);
        } else {
            /*如果等于0,则说明e=root.date 即查询成功*/
            return true;
        }
    }
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34

3.3 检查是否平衡的方法

很简单,只需要递归的查看所有节点,判断是否存在的节点的左右子节点高度差绝对值是否大于1的情况就能判断了,如果存在,那么返回false表示不是平衡二叉树,不存在就返回true表示是平衡二叉树。

    /**
     * 保存是否平衡的标志
     */
    private boolean balance = true;

    /**
     * 检查是否是平衡二叉树的方法,当然也可以debug看,如果你不嫌麻烦……
     *
     * @return true 是 ;false 否
     */
    public boolean checkBalance() {
        checkBalance(root);
        boolean balanceNow=balance;
        balance=true;
        return balanceNow;
    }

    /**
     * 递归检查是否平衡,实际上这里采用了后序遍历,即左子节点-右子节点-根节点的方法递归遍历检查
     *
     * @param root 根节点
     * @return 节点的高度
     */
    private int checkBalance(BinaryTreeNode<E> root) {
        if (root == null) {
            return -1;
        }
        //返回左子树的高度
        int hl = checkBalance(root.left);
        //返回右子树的高度
        int hr = checkBalance(root.right);
        //如果root的左右子树高度差绝对值大于1,或者checkBalance和getHeight方法获取的左/右子树高度不一致,那么算作不平衡
        if (Math.abs(getHeight(root.left) - getHeight(root.right)) > 1 ||
                getHeight(root.left) != hl || getHeight(root.right) != hr) {
            balance = false;
        }
        return getHeight(root);
    }
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35
  • 36
  • 37
  • 38

3.4 插入的方法

平衡二叉树和二叉排序树的最大区别就是在插入和删除的时候了。我们已经讨论过插入之后的4种出现平衡问题的特殊情况,这里不再赘述,下面看代码具体如何实现:

   /**
     * 插入,开放给外部使用的api
     *
     * @param e 要插入的元素
     */
    public void insert(E e) {
        //返回root,但此时新的节点可能已经被插入进去了
        root = insert(e, root);
    }

    /**
     * 插入,开放给外部使用的api
     *
     * @param es 要插入的元素的数组,注意,数组元素的顺序存储的位置将会影响二叉排序树的生成
     */
    public void insert(E[] es) {
        //返回root,但此时新的节点可能已经被插入进去了
        for (E e : es) {
            root = insert(e, root);
        }

    }

    /**
     * 插入,内部调用的方法,先从根节点开始递归查找要插入的位置,然后插入
     * 大部分代码都和排序二叉树的相似,区别就是在插入之后,会调用尝试重平衡的方法rebalance
     *
     * @param e    要插入的数据
     * @param root 节点
     * @return 原节点重平衡之后的节点或者新插入的节点
     */
    private BinaryTreeNode<E> insert(E e, BinaryTreeNode<E> root) {
        /*没有查找到,那么直接构建新的节点返回,将会在上一层方法中被赋值给其父节点的某个引用,这个插入的位置肯定是该遍历路径上的最后一点
         * 即插入的元素节点肯定是属于叶子节点*/
        if (root == null) {
            size++;
            return new BinaryTreeNode<>(e);
        }
        /*调用比较的方法*/
        int i = compare(e, root.data);
        /*如果大于0,则说明e>root.date 继续查询右子树*/
        if (i > 0) {
            //重新赋值
            root.right = insert(e, root.right);
            /*如果小于0,则说明e<root.date 继续查询左子树*/
        } else if (i < 0) {
            //重新赋值
            root.left = insert(e, root.left);
        } else {
            /*如果等于0,则说明e=root.date 即存在节点 什么都不做*/
        }
        /*insert递归插入之后,在返回时,会调用重新平衡并且设置高度的方法 尝试重平衡root根节点  而不是像排序二叉树一样简单的返回root
         *从新插入节点的父节点一直向上回溯直到根节点,尝试寻找最小不平衡树,找到之后会进行平衡,返回返回平衡之后的树,.*/
        return rebalance(root);
    }

    /**
     * 重平衡的方法
     * 1)	在节点X的左孩子节点的左子树中插入元素,简称LL 右旋
     * 2)	在节点X的左孩子节点的右子树中插入元素,简称LR 左-右双旋
     * 3)	在节点X的右孩子节点的左子树中插入元素,简称RL 左旋
     * 4)	在节点X的右孩子节点的右子树中插入元素,简称RR 右-左双旋
     *
     * @param root 树的根节点,无论是否是最小不平衡树,都是走这个方法
     * @return 平衡之后的树的根节点
     */
    private BinaryTreeNode<E> rebalance(BinaryTreeNode<E> root) {
        /*1、如果节点为null,直接返回null*/
        if (root == null) {
            return null;
        }
        /*2、开始旋转*/
        /*2.1、如果左子树的高度减去右子树的高度值大于1,说明左子树的高度大于右子树的高度至少2层,可能是情况1、2 继续判断*/
        if (getHeight(root.left) - getHeight(root.right) > 1) {
            /*如果左子节点的左子节点高度大于左子节点的右子节点高度,那说明是情况1,否则是情况2*/
            if (getHeight(root.left.left) >= getHeight(root.left.right)) {
                /*2.1.1、右旋*/
                root = rotateRight(root);
            } else {
                /*2.1.2、左-右双旋*/
                root = rotateLeftAndRight(root);
            }
            /*2.2、如果右子树的高度减去左子树的高度值大于1,说明右子树的高度大于左子树的高度至少2层,可能是情况3、4 继续判断*/
        } else if (getHeight(root.right) - getHeight(root.left) > 1) {
            /*如果右子节点的右子节点高度大于右子节点的左子节点高度,那说明是情况4,否则是情况3*/
            if (getHeight(root.right.right) >= getHeight(root.right.left)) {
                /*2.2.1、左旋*/
                root = rotateLeft(root);
            } else {
                /*2.2.2、右-左双旋*/
                root = rotateRightAndLeft(root);
            }
        }
        /*3、到这一步,说明旋转完毕,或者不需要旋转,但是都需要重新计算高度,高度为左/右子树高度最大值+1*/
        root.height = Math.max(getHeight(root.left), getHeight(root.right)) + 1;
        return root;
    }


    /**
     * 右旋
     * 通解:右旋之后,k2成为根节点,k1成为k2的右子节点,k2的右子树2成为k1的左子树
     *
     * @param k1 需要旋转的最小不平衡树根节点
     * @return k2 旋转后的最小不平衡树根节点, 已经转换为平衡二叉树
     */
    private BinaryTreeNode<E> rotateRight(BinaryTreeNode<E> k1) {
        //获取k2,k2是k1的左子节点
        BinaryTreeNode<E> k2 = k1.left;
        //k2的右子树成为k1的左子树
        k1.left = k2.right;
        //k1成为k2的右子节点
        k2.right = k1;
        //k1的高度等于k1的左或者右子树的高度的最大值+1;
        k1.height = Math.max(getHeight(k1.left), getHeight(k1.right)) + 1;
        //k2的高度等于k2的左子节点和k1的高度(此时k1就是k2的右子节点)的最大值+1
        k2.height = Math.max(getHeight(k2.left), k1.height) + 1;
        //返回k2,k2成为根节点
        return k2;
    }

    /**
     * 左旋 很简单,实际上就是右旋的镜像
     * 通解:左旋之后,k2成为根节点,k1成为k2的左子节点,k2的左子树2成为k1的右子树
     *
     * @param k1 需要旋转的最小不平衡树根节点
     * @return k2 旋转后的最小不平衡树根节点, 已经转换为平衡二叉树
     */
    private BinaryTreeNode<E> rotateLeft(BinaryTreeNode<E> k1) {
        //获取k2,k2是k1的右子节点
        BinaryTreeNode<E> k2 = k1.right;
        //k2的左子树成为k1的右子树
        k1.right = k2.left;
        //k1成为k2的左子节点
        k2.left = k1;
        //k1的高度等于k1的左或者右子树的高度的最大值+1;
        k1.height = Math.max(getHeight(k1.left), getHeight(k1.right)) + 1;
        //k2的高度等于k2的右子节点和k1的高度(此时k1就是k2的左子节点)的最大值+1
        k2.height = Math.max(getHeight(k2.right), k1.height) + 1;
        //返回k2,k2成为根节点
        return k2;
    }

    /**
     * 右-左双旋
     * 通解:将k3当作新的根节点,并且先使得k2右旋成为k3的右子树,然后k1左旋成为k3的左子树,并且左子树2成为k1的右子树,右子树2成为k2的左子树
     *
     * @param k1 需要旋转的最小不平衡树根节点
     * @return 旋转后的最小不平衡树根节点, 已经转换为平衡二叉树
     */
    private BinaryTreeNode<E> rotateRightAndLeft(BinaryTreeNode<E> k1) {
        /*1、先对k1的右子节点k2进行右旋,返回右旋之后的根节点k3,然后使得成为k3成为的k1的左子树*/
        k1.right = rotateRight(k1.right);
        /*2、然后对k1进行左旋,成为k3的左子树,返回的根节点就是k3,即返回旋转之后的根节点*/
        return rotateLeft(k1);
    }

    /**
     * 左-右双旋 很简单,实际上就是右-左双旋的镜像
     * 通解: 将k3当作新的根节点,并且先使得k2左旋成为k3的左子树,然后k1右旋成为k3的右子树,并且左子树2成为k2的右子树,右子树2成为k1的左子树
     *
     * @param k1 需要旋转的最小不平衡树根节点
     * @return 旋转后的最小不平衡树根节点, 已经转换为平衡二叉树
     */
    private BinaryTreeNode<E> rotateLeftAndRight(BinaryTreeNode<E> k1) {
        /*1、先对k1的左子节点k2进行左旋,返回左旋之后的根节点k3,然后使得成为k3成为的k1的左子树*/
        k1.left = rotateLeft(k1.left);
        /*2、然后对k1进行右旋,成为k3的右子树,返回的根节点就是k3,即返回旋转之后的根节点*/
        return rotateRight(k1);
    }
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35
  • 36
  • 37
  • 38
  • 39
  • 40
  • 41
  • 42
  • 43
  • 44
  • 45
  • 46
  • 47
  • 48
  • 49
  • 50
  • 51
  • 52
  • 53
  • 54
  • 55
  • 56
  • 57
  • 58
  • 59
  • 60
  • 61
  • 62
  • 63
  • 64
  • 65
  • 66
  • 67
  • 68
  • 69
  • 70
  • 71
  • 72
  • 73
  • 74
  • 75
  • 76
  • 77
  • 78
  • 79
  • 80
  • 81
  • 82
  • 83
  • 84
  • 85
  • 86
  • 87
  • 88
  • 89
  • 90
  • 91
  • 92
  • 93
  • 94
  • 95
  • 96
  • 97
  • 98
  • 99
  • 100
  • 101
  • 102
  • 103
  • 104
  • 105
  • 106
  • 107
  • 108
  • 109
  • 110
  • 111
  • 112
  • 113
  • 114
  • 115
  • 116
  • 117
  • 118
  • 119
  • 120
  • 121
  • 122
  • 123
  • 124
  • 125
  • 126
  • 127
  • 128
  • 129
  • 130
  • 131
  • 132
  • 133
  • 134
  • 135
  • 136
  • 137
  • 138
  • 139
  • 140
  • 141
  • 142
  • 143
  • 144
  • 145
  • 146
  • 147
  • 148
  • 149
  • 150
  • 151
  • 152
  • 153
  • 154
  • 155
  • 156
  • 157
  • 158
  • 159
  • 160
  • 161
  • 162
  • 163
  • 164
  • 165
  • 166
  • 167
  • 168
  • 169
  • 170

3.4.1 测试

AvlTree<Integer> avlTree = new AvlTree<>();
Integer[] es = new Integer[]{3, 2, 1, 4, 5, 6, 7, 16, 15, 14, 13, 12, 11, 10, 8, 9};
//批量插入
avlTree.insert(es);
//中序遍历输出
System.out.println(avlTree.toInorderTraversalString());
//检查是否平衡
System.out.println(avlTree.checkBalance());
//数量
System.out.println(avlTree.size());

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11

在insert之后打上断点,Debug,可以看到avlTree的数据结构和在实现原理中最终的结构是一致的。

在这里插入图片描述

在这里插入图片描述

3.5 查找最大值和最小值

很简单,最左边的节点一定是最小的,最右边的节点一定是最大的。因此查找最小的节点只需要向左递归查找,查找最大的节点只需要向右递归查找。

/**
 * 查找最小的节点
 *
 * @param root 根节点
 * @return 最小的节点
 */
private BinaryTreeNode<E> findMin(BinaryTreeNode<E> root) {
    if (root == null) {
        return null;
        /*如果该节点没有左右子节点,那么该节点就是最小的节点,返回*/
    } else if (root.left == null) {
        return root;
    }
    /*如果该节点存在左子节点,那么继续向左递归查找*/
    return findMin(root.left);
}

/**
 * 查找最大的节点
 *
 * @param root 根节点
 * @return 最大的节点
 */
private BinaryTreeNode<E> findMax(BinaryTreeNode<E> root) {
    if (root == null) {
        return null;
        /*如果该节点没有右子节点,那么该节点就是最大的节点,返回*/
    } else if (root.right == null) {
        return root;
    }
    /*如果该节点存在右子节点,那么继续向右递归查找*/
    return findMax(root.right);
}

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34

3.6 删除的方法

平衡二叉树节点的删除同样可能导致产生不平衡的状态,因此同样在二叉排序树的删除代码的基础上,删除元素之后需要在删除节点之上进行回溯直到根节点,尝试找出最小不平衡树来进行重平衡。其平衡的方法是和插入的时候是一样的。

/**
 * 删除,开放给外部使用的api
 *
 * @param e 要删除的元素
 */
public void delete(E e) {
    //返回root,但此时可能有一个节点已经被删除了
    root = delete(e, root);
}

/**
 * 删除,内部调用的方法,删除分为三种情况: 1、该节点没有子节点 2、该字节仅有一个子节点 3、该节点具有两个子节点
 *
 * @param e    要删除的数据
 * @param root (子)树根节点
 * @return 该根节点重平衡之后的节点
 */
private BinaryTreeNode<E> delete(E e, BinaryTreeNode<E> root) {
    /*没有查找到,那么什么都不做*/
    if (root == null) {
        return null;
    }
    /*调用比较的方法*/
    int i = compare(e, root.data);
    /*如果大于0,则说明e>root.date 继续查询右子树*/
    if (i > 0) {
        //重新赋值
        root.right = delete(e, root.right);
        /*如果小于0,则说明e<root.date 继续查询左子树*/
    } else if (i < 0) {
        //重新赋值
        root.left = delete(e, root.left);
    } else {
        /*如果等于0,则说明e=root.date 即查询成功 开始执行删除*/
        /*如果两个子节点都不为null*/
        if (root.left != null && root.right != null) {
            /*递归查找最小的节点,然后递归删除*/
            root.data = findMin(root.right).data;
            root.right = delete(root.data, root.right);
        } else {
            /*如果一个子节点不为null,则返回该子节点;或者两个子节点都为null,则返回null
             * 此时该root节点已经被"绕过了"*/
            root = (root.left != null) ? root.left : root.right;
            --size;
        }
    }
    /*和二叉排序树直接返回节点不同的是,删除操作完成之后将会调用该方法,从被删除节点回溯至根节点,对节点进行重平衡,然后才返回平衡后的节点*/
    return rebalance(root);
}

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35
  • 36
  • 37
  • 38
  • 39
  • 40
  • 41
  • 42
  • 43
  • 44
  • 45
  • 46
  • 47
  • 48
  • 49
  • 50

3.6.1 测试

针对上面插入的平衡二叉树进行删除:

System.out.println("======>首先删除5  此时没有影响,不需要重平衡");
avlTree.delete(5);
//检查是否平衡
System.out.println(avlTree.checkBalance());
//中序遍历输出
System.out.println(avlTree.toInorderTraversalString());


System.out.println("======>再次删除6  此时节点4的BF为2 需要右旋重平衡");
avlTree.delete(6);
//检查是否平衡
System.out.println(avlTree.checkBalance());
//中序遍历输出
System.out.println(avlTree.toInorderTraversalString());

System.out.println("======>再次删除11  由于该节点拥有左右子树,实际上删除的是该节点的右子树的最小节点12,然后将12的值赋值给11的节点,并导致节点12的原父节点11不平衡,需要重平衡");
avlTree.delete(11);
//检查是否平衡
System.out.println(avlTree.checkBalance());
//中序遍历输出
System.out.println(avlTree.toInorderTraversalString());

System.out.println("======>再次删除7  由于该节点拥有左右子树,实际上删除的是该节点的右子树的最小节点8,然后将8的值赋值给7的节点,并导致节点8的原父节点9不平衡,需要重平衡");
avlTree.delete(7);
//检查是否平衡
System.out.println(avlTree.checkBalance());
//中序遍历输出
System.out.println(avlTree.toInorderTraversalString());

System.out.println("======>再次删除9、12  此时不需要重平衡");
avlTree.delete(9);
avlTree.delete(12);
//检查是否平衡
System.out.println(avlTree.checkBalance());
//中序遍历输出
System.out.println(avlTree.toInorderTraversalString());

System.out.println("======>最后删除8  由于该节点拥有左右子树,实际上删除的是该节点的右子树的最小节点10节点,然后将10的值赋值给8的节点,并导致节点10的原父节点13不平衡,需要重平衡");
avlTree.delete(8);
//检查是否平衡
System.out.println(avlTree.checkBalance());
//中序遍历输出
System.out.println(avlTree.size());
System.out.println(avlTree.toInorderTraversalString());

  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16
  • 17
  • 18
  • 19
  • 20
  • 21
  • 22
  • 23
  • 24
  • 25
  • 26
  • 27
  • 28
  • 29
  • 30
  • 31
  • 32
  • 33
  • 34
  • 35
  • 36
  • 37
  • 38
  • 39
  • 40
  • 41
  • 42
  • 43
  • 44
  • 45

在进行上面的一系列删除之后,树结构会变成如下形状:

在这里插入图片描述

4 平衡二叉树的总结

平衡二叉树是基于二叉排序树的,但是由于其必须保持平衡的特性,因此其编码难度比二叉排序树的编码难度更高,不过如果我们搞懂了其旋转的原理,那么实现起来还是比较简单的。

如果我们需要查找的集合本身没有顺序,在频繁查找的同时也需要经常的插入和删除操作,显然我们需要构建一棵二叉排序树,但是不平衡的二叉排序树,极端情况下可能退化为链表,查找效率是非常低的,因此我们需要在构建时,就让这棵二叉排序树是动态的转换为平衡二叉树,此时我们的查找时间复杂度就为O(logn),而插入和删除也为O(logn)。这显然是比较理想的一种动态查找表算法。

本文介绍了平衡二叉树的原理以及Java代码的完全实现,要想搞懂平衡二叉树需要先搞懂二叉排序树:二叉排序树的详解以及Java代码的完全实现。而搞懂平衡二叉树又是搞懂红黑树的基础,后续文章我们将会介绍红黑树的概念以及Java代码的完全实现!

相关参考:

  1. 《算法》
  2. 《数据结构与算法分析》
  3. 《大话数据结构》

如果有什么不懂或者需要交流,可以留言。另外希望点赞、收藏、关注,我将不间断更新各种Java学习博客!

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/AllinToyou/article/detail/632142
推荐阅读
相关标签
  

闽ICP备14008679号