平衡二叉树（AVL树）

一、平衡二叉树概述

1.1 什么是平衡二叉树

平衡二叉树也叫 AVL 树。平衡二叉树是具有以下特点的二叉查找树：它是一棵空树或它的左右两个子树的高度差的绝对值不超过 1， 并且左右两个子树都是一棵平衡二叉树。

图一 平衡二叉树

图二 非平衡二叉树

1.2 为什么要有平衡二叉树

在上一篇文章 二叉排序树 中对二叉排序树做了介绍。

我们知道二叉排序树是结合了数组和链表的优点的一种数据结构，它在增删节点效率还不错的同时还有具有很好的查询效率。

但是对于下面这样的二叉排序树：

它的左子树为空，从形式上看更像是一个链表。虽然它在插入、删除速度上依然很快，但是查询的速度却和链表一样甚至更慢（因为还需要判断有无左子树）。这样的二叉排序树显然丧失了最大的优点。

我们可以对它改造，将其改造成平衡二叉树：

可以看到，在不改变二叉树元素的前提下，新的二叉树的查询速度得到了有效提升。

这就是平衡二叉树的意义。

二、平衡二叉树的操作

2.1 左旋转

2.1.1 需要左旋转的情况

当根节点的右子树高度减去左子树高度大于 1 时，需要左旋二叉树。

对于下面这样的二叉树：

我们可以看出来，它的左子树的高度比右子树的高度少 2。

我们可以通过将其左旋转使其变成下面这样的平衡二叉树：

2.1.2 左旋转步骤

左旋转的基本步骤如下：

1. 首先创建一个新节点 ，该节点的值设为根节点的值；
2. 新节点的左指针指向根节点的左子树；
3. 新节点的右指针指向根节点的右子节点的左子树；
4. 将根节点的值设置为根节点的右子节点的值；
5. 根节点的左指针指向新节点；
6. 根节点的右指针指向其右子节点的右子树。

左旋转过程图解如下：

2.1.3 左旋转代码实现

左旋转的代码实现如下，该函数是节点类的成员函数：

// 以当前节点为根节点将二叉树左旋
public void leftRotate(){
   
    // 1. 新建一个节点，值为根节点的值
    AVLNode newNode = new AVLNode(this.value);
    // 2. 新节点的左指针指向根节点的左子树
    newNode.left = this.left;
    // 3. 新节点的右指针指向根节点的右子树的左子树
    newNode.right = this.right.left;
    // 4. 将根节点的值改为其右子节点的值
    this.value = this.right.value;
    // 5. 根节点的左指针指向新节点
    this.left = newNode;
    // 6. 根节点的右指针指向其右子节点的右子节点
    this.right = this.right.right;
}

2.2 右旋转

2.2.1 需要右旋转的情况

当根节点的左子树高度减去右子树高度大于 1 时，需要右旋二叉树。

如下图图左的二叉排序树，其根节点左子树的高度比右子树大 2，将其右旋转之后，新的二叉树的搜索效率有效提升。

2.2.2 右旋转步骤

将二叉树右旋转基本步骤如下：

1. 首先创建一个新节点，新节点的值设为根节点的值；
2. 新节点的右指针指向根节点的右子树；
3. 新节点的左指针指向根节点的左子节点的右子树；
4. 将根节点的值设为其左子节点的值；
5. 根节点的左指针指向其左子节点的左子树；
6. 根节点的右指针指向新节点。

二叉树右旋转过程与左旋转类似，可以参照左旋转的步骤图来理解。

2.2.3 右旋转代码实现

右旋转的代码实现如下，该函数是节点类的成员函数：

// 以当前节点为根节点将二叉树右旋
public void rightRotate(){
   
    // 1. 新建一个节点，值为根节点的值
    AVLNode newNode = new AVLNode(this.value);
    // 2. 将新节点的右指针指向根节点的右子树
    newNode.right = this.right;
    // 3. 将新节点的左指针指向根节点的左子树的右子树
    newNode.left = this.left.right;
    // 4. 将根节点的值改为其左子节点的值
    this.value = this.left.value;
    // 5. 将根节点的左指针指向其左子节点的左子树
    this.left =