Java尚硅谷数据结构与算法学习记录(14)-AVL树

平衡二叉树(AVL树)

平衡二叉树也叫平衡二叉搜索树（Self-balancing binary search tree）又被称为AVL树， 可以保证查询效率较高。
具有以下特点：它是一 棵空树或它的左右两个子树的高度差的绝对值不超过1，并且左右两个子树都是一棵平衡二叉树。平衡二叉树的常用实现方法有红黑树、AVL、替罪羊树、Treap、伸展树等。

应用实例：左旋转

代码实现：

//左旋转方法
    private void leftRotate() {
   
        //创建一个新的节点 值等于当前根节点的值
        Node newNode = new Node(value);
        //把新节点的左子树设置为当前节点的左子树
        newNode.left = left;
        //把新节点的右子树设置为当前节点的右子树的左子树
        newNode.right = right.left;
        //把当前节点的值换为右子节点的值
        value = right.value;
        //把当前节点的右子树设置为右子树的右子树
        right = right.right;
        //把当前节点的左子树设置为新节点
        left = newNode;
    }

应用实例：右旋转

代码实现：

//右旋转方法
    private void rightRotate() {
   
        //创建一个新的节点 值等于当前根节点的值
        Node newNode = new Node(value);
        //把新节点的右子树设置为当前节点的右子树
        newNode.right = right;
        //把新节点的左子树设置为当前节点的左子树的右子树
        newNode.left = left.right;
        //把当前节点的值换为左子节点的值
        value = left.value;
        //把当前节点的左子树设置为左子树的左子树
        left = left.left;
        //把当前节点的右子树设置为新节点
        right = newNode;
    }

应用实例：双旋转
数列int[] arr = { 10, 11, 7, 6, 8, 9 }; int[] arr = {2,1,6,5,7,3}; 运行原来的代码可以看到，并没有转成 AVL树.。当经过一次旋转无法使二叉树变成平衡二叉树时，需要进行双旋转。

分析图示：

当符合左旋转的条件时：如果它的右子树的左子树的高度大于它的右子树的右子树的高度 则先对当前节点的右节点进行右旋转，再对当前节点进行左旋转。
当符合右旋转的条件时：如果它的左子树的右子树高度大于它的左子树的左子树的高度，则先对当前节点的左节点进行左旋转，再对当前节点进行右旋转。

代码实现：

//在add方法中添加
//当添加完一个节点后，如果（右子树的高度-左子树的高度）>1 左旋转
        if ((rightHeight() - leftHeight()) > 1) {
   
            //如果它的右子树的左子树的高度大于它的右子树的右子树的高度
            if (right != null && right.leftHeight() > right.rightHeight()) {
   
                //先对它的右子节点进行右旋转
                right.rightRotate();
                //再对当前结点进行左旋转
                leftRotate();
            } else {
   
                leftRotate();
            }
            return;//注意一定要return  因为平衡好后不必再往下走了 （或者可以跟下面的做else）！！
        }
        //当添加完一个节点后，如果（左子树的高度-右子树的高度）>1 右旋转
        if ((leftHeight() - rightHeight()) > 1) {
   
            //如果它的左子树的右子树高度大于它的左子树的左子树的高度
            if (left != null && left.rightHeight() > left.leftHeight()) {
   
                //先对当前节点的左节点进行左旋转
                left.leftRotate();
                //在对当前节点进行右旋转
                rightRotate();
            } else {
   
                rightRotate();
            }
        }

全部代码：

package com.datastructures.avl;

public class AVLTreeDemo {
   
    public static void main(String[] args) {
   
        //int [] arr={4,3,6,5,7,8};
        //int[] arr = {10, 12, 8, 9, 7, 6};
        int[] arr={
   10, 11, 7, 6, 8, 9};
        AVLTree avlTree = new AVLTree();
        for (int i = 0; i < arr.leng