AVL树(AVL Tree)

        AVL树是一种自平衡的二叉搜索树,它具有以下特性:1. 每个节点的左子树和右子树的高度最多相差1(平衡性)。 2. 每个节点存储一个平衡因子(Balance Factor),即其左子树的高度减去右子树的高度。 
        AVL树通过旋转操作来保持树的平衡,包括左旋转、右旋转、左右旋转和右左旋转等操作。在插入和删除节点时,AVL树会根据节点的平衡因子进行调整,以保持树的平衡性。

        如果需要一种查询高效且有序的数据结构,而且数据的个数为静态的(即不会改变),可以考虑AVL树。但一个结构经常修改,就不太适合。

        Java简单示例:

public class AVLTree {
    
    private Node root;

    private class Node {
        int key;
        int height;
        Node left;
        Node right;

        public Node(int key) {
            this.key = key;
            this.height = 1;
        }
    }
    // 获取节点高度
    private int height(Node node) {
        if (node == null) {
            return 0;
        }
        return node.height;
    }
    // 求两个数的最大值
    private int max(int a, int b) {
        return Math.max(a, b);
    }
    // 右旋转操作
    private Node rightRotate(Node y) {
        Node x = y.left;
        Node T2 = x.right;

        x.right = y;
        y.left = T2;

        y.height = max(height(y.left), height(y.right)) + 1;
        x.height = max(height(x.left), height(x.right)) + 1;

        return x;
    }
    // 左旋转操作
    private Node leftRotate(Node x) {
        Node y = x.right;
        Node T2 = y.left;

        y.left = x;
        x.right = T2;

        x.height = max(height(x.left), height(x.right)) + 1;
        y.height = max(height(y.left), height(y.right)) + 1;

        return y;
    }
    // 获取平衡因子
    private int getBalance(Node node) {
        if (node == null) {
            return 0;
        }
        return height(node.left) - height(node.right);
    }
    // 插入节点
    public void insert(int key) {
        root = insertNode(root, key);
    }

    private Node insertNode(Node node, int key) {
        // 实现插入逻辑
        if (node == null) {
            return new Node(key);
        }

        if (key < node.key) {
            node.left = insertNode(node.left, key);
        } else if (key > node.key) {
            node.right = insertNode(node.right, key);
        } else {
            return node;
        }

        node.height = 1 + max(height(node.left), height(node.right));

        int balance = getBalance(node);

        if (balance > 1 && key < node.left.key) {
            return rightRotate(node);
        }
        if (balance < -1 && key > node.right.key) {
            return leftRotate(node);
        }
        if (balance > 1 && key > node.left.key) {
            node.left = leftRotate(node.left);
            return rightRotate(node);
        }
        if (balance < -1 && key < node.right.key) {
            node.right = rightRotate(node.right);
            return leftRotate(node);
        }

        return node;
    }
    // 删除节点
    public void delete(int key) {
        root = deleteNode(root, key);
    }
    private Node deleteNode(Node node, int key) {
        // 实现删除逻辑
        if (node == null) {
            return null;
        }

        if (key < node.key) {
            node.left = deleteNode(node.left, key);
        } else if (key > node.key) {
            node.right = deleteNode(node.right, key);
        } else {
            if (node.left == null || node.right == null) {
                Node temp = null;
                if (temp == node.left) {
                    temp = node.right;
                } else {
                    temp = node.left;
                }

                if (temp == null) {
                    temp = node;
                    node = null;
                } else {
                    node = temp;
                }
            } else {
                Node temp = minValueNode(node.right);
                node.key = temp.key;
                node.right = deleteNode(node.right, temp.key);
            }
        }

        if (node == null) {
            return null;
        }

        node.height = 1 + max(height(node.left), height(node.right));

        int balance = getBalance(node);

        if (balance > 1 && getBalance(node.left) >= 0) {
            return rightRotate(node);
        }
        if (balance < -1 && getBalance(node.right) <= 0) {
            return leftRotate(node);
        }
        if (balance > 1 && getBalance(node.left) < 0) {
            node.left = leftRotate(node.left);
            return rightRotate(node);
        }
        if (balance < -1 && getBalance(node.right) > 0) {
            node.right = rightRotate(node.right);
            return leftRotate(node);
        }

        return node;
    }
    // 寻找子树中最小值的节点
    private Node minValueNode(Node node) {
        Node current = node;
        while (current.left != null) {
            current = current.left;
        }
        return current;
    }
    // 中序遍历
    public void inorder() {
        inorderTraversal(root);
    }

    private void inorderTraversal(Node node) {
        if (node != null) {
            inorderTraversal(node.left);
            System.out.print(node.key + " ");
            inorderTraversal(node.right);
        }
    }
    // 查询节点
    public Node search(int key) {
        return searchNode(root, key);
    }

    private Node searchNode(Node node, int key) {
        if (node == null || node.key == key) {
            return node;
        }
        if (key < node.key) {
            return searchNode(node.left, key);
        } else {
            return searchNode(node.right, key);
        }
    }
}

        实战Java代码

import java.util.ArrayList;
import java.util.List;

/**
 * @author xiarg
 * @date 2024/4/19 14:17
 */
public class AVLTree {

    private static class TreeNode {
        int val;
        // 平衡因子 = 右子树高度 - 左子树的高度
        int bf;
        TreeNode left;
        TreeNode right;
        TreeNode parent;

        TreeNode(int val) {
            this.val = val;
        }
    }

    private TreeNode root;

    /**
     * 插入操作
     *   1. 先把数据插入二叉树中
     *   2. 根据平衡因子对树进行平衡调整
     * @return 无法插入重复元素
     */
    public boolean insert(int val) {
        TreeNode node = new TreeNode(val);
        if(root == null) {
            root = node;
            return true;
        }

        TreeNode parent = null;
        TreeNode cur = root;
        while(cur != null) {
            if(val == cur.val){
                return false;
            }else if(val < cur.val) {
                parent = cur;
                cur = parent.left;
            }else{
                parent = cur;
                cur = parent.right;
            }
        }
        // 此时cur=null,插入 val
        if(val < parent.val) {
            parent.left = node;
        }else {
            parent.right = node;
        }
        node.parent = parent;
        cur = node;

        /**
         * 自底向上调节平衡因子
         *   因为 bf = rightH - leftH, 所以当某个子树插入一个节点不引起左右子树高度变化时,就不需要继续向上调整平衡因子了
         *   也就是,当某个子树节点从 (-1 / 1) -> 0 的时候
         */
        while(parent != null) {
            // 一直向上调节
            if(cur == parent.left) {
                // cur 是 parent 的左孩子
                parent.bf --;
            }else {
                parent.bf ++;
            }

            if(parent.bf == 0) {
                // 已经平衡了
                break;
            }else if(parent.bf == -1 || parent.bf == 1) {
                // 继续向上
                cur = parent;
                parent = cur.parent;
            }else {
                // parent.bf = -2/2, 该子树不平衡了,应该旋转, 有以下4种情况:
                if(parent.bf == -2 && cur.bf == -1) {
                    // LL型,右单旋
                    rotateR(parent);
                }else if(parent.bf == 2 && cur.bf == 1) {
                    // RR型,左单旋
                    rotateL(parent);
                }else if(parent.bf == -2 && cur.bf == 1) {
                    // LR型,左右双旋
                    rotateLR(parent);
                }else {
                    // parent.bf == 2 && cur.bf == -1
                    // RL型,右左双旋
                    rotateRL(parent);
                }
                // 旋转完就一定平衡了
                break;
            }
        }
        return true;
    }
    // 左单旋
    private void rotateL(TreeNode p) {
        // p 的母节点
        TreeNode pp = p.parent;
        // p 的右孩子
        TreeNode subR = p.right;
        // subR 的左孩子,可能不存在
        TreeNode subRL = subR.left;

        // subR 提上去
        if(pp == null) {
            this.root = subR;
        }else if(pp.left == p) {
            pp.left = subR;
        }else {
            // pp.right == parent
            pp.right = subR;
        }
        subR.parent = pp;
        // p 作为 subR 的左孩子
        subR.left = p;
        p.parent = subR;
        // p 与 subRL 连接
        p.right = subRL;
        if(subRL != null) {
            subRL.parent = p;
        }

        // 经过旋转最终只会改变 parent 和 subR 的平衡因子
        // 而且无论 subRL 是否存在,最终都是 0
        subR.bf = 0;
        p.bf = 0;
    }
    // 右单旋
    private void rotateR(TreeNode p) {
        // p 的父节点
        TreeNode pp = p.parent;
        // p 的左孩子
        TreeNode subL = p.left;
        // subL 的右孩子,可能不存在
        TreeNode subLR = subL.right;

        if(pp == null) {
            this.root = subL;
        }else if(pp.left == p) {
            pp.left = subL;
        }else {
            pp.right = subL;
        }
        subL.parent = pp;
        subL.right = p;
        p.parent = subL;
        p.left = subLR;
        if(subLR != null) {
            subLR.parent = p;
        }
        // 经过旋转最终只会改变 parent 和 subL 的平衡因子
        // 而且无论 subLR 是否存在,最终都是 0
        subL.bf = 0;
        p.bf = 0;
    }
    // 左右双旋
    private void rotateLR(TreeNode p) {
        TreeNode subL = p.left;
        TreeNode subLR = subL.right;
        int bf = subLR.bf; // 可能为 0、1、-1

        rotateL(subL);
        rotateR(p);

        subLR.bf = 0;
        if(bf == 1) {
            p.bf = 0;
            subL.bf = -1;
        }else if(bf == -1) {
            subL.bf = 0;
            p.bf = 1;
        }
        // 当 bf == 0 时,就不需要再手动改平衡因子了
    }
    // 右左双旋
    private void rotateRL(TreeNode p) {
        TreeNode subR = p.right;
        TreeNode subRL = subR.left;
        int bf = subRL.bf;

        rotateR(subR);
        rotateL(p);

        subRL.bf = 0;
        if(bf == -1) {
            p.bf = 0;
            subR.bf = 1;
        }else if(bf == 1) {
            subR.bf = 0;
            p.bf = -1;
        }
    }

    /**
     * 查找操作
     */
    public boolean contains(int val) {
        return contains(root, val);
    }
    // 在当前树中查找元素
    private boolean contains(TreeNode root, int val) {
        if(root == null) {
            return false;
        }
        if(val == root.val) {
            return true;
        }else if(val < root.val) {
            return contains(root.left, val);
        }else {
            return contains(root.right, val);
        }
    }

    /** 打印二叉树, 中序遍历的结果 **/
    @Override
    public String toString() {
        List<Integer> list = new ArrayList<>();
        inOrder(list, root);
        return list.toString() + "\n" + "是否平衡: " + isBalance(this.root);
    }
    // 中序遍历
    private void inOrder(List<Integer> list, TreeNode root) {
        if(root == null){
            return;
        }
        inOrder(list, root.left);
        list.add(root.val);
        inOrder(list, root.right);
    }

    /**
     * 检验二叉树是否平衡
     *   每个节点子树高度差的绝对值不超过1(注意节点中如果没有平衡因子)
     *   节点的平衡因子是否计算正确
     */
    private boolean isBalance(TreeNode root) {
        if(root == null) {
            return true;
        }
        int leftH = height(root.left);
        int rightH = height(root.right);
        if(rightH - leftH != root.bf) {
            System.err.println("平衡因子异常!");
            return false;
        }
        return Math.abs(root.bf) < 2 && isBalance(root.left) && isBalance(root.right);
    }
    // 计算二叉树的高度
    private int height(TreeNode root) {
        if(root == null) {
            return 0;
        }
        return 1 + Math.max(height(root.left), height(root.right));
    }
}

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
判断一棵二叉是否为 AVL ,需要满足以下两个条件: 1. 该二叉的所有子都是 AVL ; 2. 该二叉的任意两个子高度差的绝对值不超过 1。 其中,子的高度指的是该子的根节点到叶子节点的最长路径长度。 因此,判断一棵二叉是否为 AVL ,需要对该二叉进行递归遍历,依次判断每个子是否为 AVL ,并计算每个子的高度,最后再判断任意两个子高度差的绝对值是否不超过 1。如果满足以上两个条件,则该二叉AVL ,否则不是 AVL 。 具体实现过程可以参考以下代码(假设二叉的节点结构为 Node,其包含 left、right、height 三个属性): ```python def is_avl_tree(root): # 辅助函数,计算节点高度 def get_height(node): if not node: return 0 return node.height # 判断一棵二叉是否为 AVL def is_avl_tree_helper(node): if not node: return True # 判断左右子是否为 AVL if not is_avl_tree_helper(node.left) or not is_avl_tree_helper(node.right): return False # 计算左右子的高度差 height_diff = abs(get_height(node.left) - get_height(node.right)) if height_diff > 1: return False # 更新当前节点的高度 node.height = max(get_height(node.left), get_height(node.right)) + 1 return True # 判断根节点是否为空,为空则返回 False if not root: return False return is_avl_tree_helper(root) ``` 注意,以上代码仅为示例,具体实现可能因语言和实际情况的不同而有所差异。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值