数据结构(三)----平衡二叉树

  • 平衡二叉树 又叫AVL树

1. 性质 :

  • 可以是空树
  • 任意一个节点的左子树和右子树都是平衡二叉树 且高度之差的绝对值 <=1
    在这里插入图片描述

2.失衡的4种情况

  • 在结点的左子树的左子树插入元素,LL 插入;
  • 在结点的左子树的右子树插入元素,LR 插入;
  • 在结点的右子树的左子树插入元素,RL 插入;
  • 在结点的右子树的右子树插入元素,RR 插入;
    在这里插入图片描述

3.解决方法 代码实现

  • 定义节点类和一些基本方法
public class AVLTree {
    //当前树
    private AvlNode tree;

    //获取节点高度
    public int hight(AvlNode node) {
        return node == null ? 0 : node.hight;
    }

    //获取树的高度
    public int hight() {
        return hight(tree);
    }

    //获取高度最大值
    private int getMaxHeight(int height1, int height2) {
        return height1 > height2 ? height1 : height2;
    }

    /**
     * 中序遍历
     * 先打印左子树
     * 然后打印自身
     * 最后打印右子树
     * @param node
     */
    public void inOrder(AvlNode node){
        if(node == null ){
            return;
        }
        inOrder(node.left);
        System.out.print(node.getValue()+"->");
        inOrder(node.right);
    }

    /**
     * 前序遍历
     * 先打印自身
     * 然后打印左子树
     * 最后打印右子树
     * @param node
     */
    public void preOrder(AvlNode node){
        if(node == null){
            return;
        }
        System.out.print(node.getValue()+"->");
        preOrder(node.left);
        preOrder(node.right);
    }
    /**
     * 后续遍历
     * 先打印左子树
     * 然后打印右子树
     * 最后打印自身
     * @param node
     */
    public void postOrder(AvlNode node){
        if(node == null){
            return;
        }
        postOrder(node.left);
        postOrder(node.right);
        System.out.print(node.getValue()+"->");
    }

    //avl树节点类
    private static class AvlNode {
        private int value;
        private AvlNode left;
        private AvlNode right;
        private int hight;//定义节点高度

        public AvlNode(int value, AvlNode left, AvlNode right, int hight) {
            this.value = value;
            this.left = left;
            this.right = right;
            this.hight = hight;
        }

        public AvlNode(int value, AvlNode left, AvlNode right) {
            this(value, left, right, 0);
        }

        public int getValue() {
            return value;
        }

        public void setValue(int value) {
            this.value = value;
        }

        public AvlNode getLeft() {
            return left;
        }

        public void setLeft(AvlNode left) {
            this.left = left;
        }

        public AvlNode getRight() {
            return right;
        }

        public void setRight(AvlNode right) {
            this.right = right;
        }

        public int getHight() {
            return hight;
        }

        public void setHight(int hight) {
            this.hight = hight;
        }
    }
}
3.1**LL : 左左旋转 **

在这里插入图片描述

/**
     * LL
     *
     * @param node 失衡节点
     * @return 左旋后的根节点
     */
    public AvlNode llRotaing(AvlNode node) {
        AvlNode avl_lift = node.left;//失衡节点的左子树节点 旋转后的根节点
        //把失衡节点左子树的右子树节点挂到失衡节点的左子树上
        node.left = avl_lift.right;
        //把失衡节点作为该节点左子树的右子树
        avl_lift.right = node;
        //重新计算失衡节点和失衡节点左子树(即旋转后的根节点)旋转后的节点高度
        node.hight = getMaxHeight(hight(node.left), hight(node.right)) + 1;
        avl_lift.hight = getMaxHeight(hight(avl_lift.left), hight(avl_lift.right)) + 1;//得到左子树和右子树中最大的高度 + 1
        return avl_lift;
    }
3.2RR 右右旋转

在这里插入图片描述

 /**
     * RR
     *
     * @param node 失衡节点
     * @return 右旋后的根节点
     */
    public AvlNode rrRotaing(AvlNode node) {
        AvlNode avl_right = node.right;//失衡节点的右子树节点 旋转后的根节点
        //把失衡节点右子树的左子树节点挂到失衡节点的右子树上
        node.right = avl_right.left;
        //把失衡节点作为该节点右子树的左子树
        avl_right.left = node;
        //重新计算失衡节点和失衡节点右子树(即旋转后的根节点)旋转后的节点高度
        node.hight = getMaxHeight(hight(node.left), hight(node.right)) + 1;
        avl_right.hight = getMaxHeight(hight(avl_right.left), hight(avl_right.right)) + 1;//得到左子树和右子树中最大的高度 + 1
        return avl_right;
    }
  
3.3 LR 左右旋转

在这里插入图片描述

    /**
     * LR
     *  1. 先把失衡节点的左子树做一次rr旋转
     *  2. 把该数做一次ll旋转
     * @param node 失衡节点
     * @return
     */
    public AvlNode lrRotaing(AvlNode node){
        node.left = rrRotaing(node.left);
        AvlNode avlNode = llRotaing(node);
        return avlNode;
    }
3.4RL 右左旋转

在这里插入图片描述

    /**
     * RL
     *  1. 先把失衡节点的右子树做一次ll旋转
     *  2. 把该数做一次rr旋转
     * @param node 失衡节点
     * @return
     */
    public AvlNode rlRotaing(AvlNode node){
        node.right = llRotaing(node.right);
        AvlNode avlNode = rrRotaing(node);
        return avlNode;
    }
3.5 添加方法
    /**
     * 添加方法
     *
     * @param value
     */
    public void insert(int value){
        this.tree = insert(tree,value);
    }

    private AvlNode insert(AvlNode node,int value){
        AvlNode p;//最左边或者最右边的节点  用于判断数是RR RL LL LR的那种失衡
        if (node == null){
            node = createNode(value,0);
        }else {
            if (value > node.value){
                node.right = insert(node.right, value);
                if (Math.abs(hight(node.right) - hight(node.left)) > 1) {//失衡类型为 RR or RL
                    p = node;
                    while (p.right != null){
                        p = p.right;
                    }
                    if (value == p.value) {//说明是RR
                        node = rrRotaing(node);
                    } else {
                        node = rlRotaing(node);
                    }
                }
            }else if (value < node.value){
                node.left = insert(node.left,  value);
                if (Math.abs(hight(node.right) - hight(node.left)) > 1) {//失衡类型为 LL or LR
                    p = node;
                    while (p.left != null){
                        p = p.left;
                    }
                    if (value == p.value) {//说明是LL
                        node = llRotaing(node);
                    } else {//否则为LR
                        node = lrRotaing(node);
                    }
                }
            }
        }
        node.hight = getMaxHeight(hight(node.left), hight(node.right)) + 1;
        return node;
    }
3.5 断点测试
    public static void main(String[] args) {
        AVLTree avlTree = new AVLTree();
        avlTree.insert(10);
        avlTree.insert(8);
        avlTree.insert(3);
        avlTree.insert(12);
        avlTree.insert(9);
        avlTree.insert(4);
        avlTree.insert(5);
        avlTree.insert(7);
        avlTree.insert(1);
        avlTree.insert(11);
        avlTree.insert(17);

    }

4.完整代码

public class AVLTree {

    public static void main(String[] args) {
        AVLTree avlTree = new AVLTree();
        avlTree.insert(10);
        avlTree.insert(8);
        avlTree.insert(3);
        avlTree.insert(12);
        avlTree.insert(9);
        avlTree.insert(4);
        avlTree.insert(5);
        avlTree.insert(7);
        avlTree.insert(1);
        avlTree.insert(11);
        avlTree.insert(17);

    }

    //根节点
    private AvlNode tree;

    private int i;

    /**
     * 添加方法
     *
     * @param value
     */
    public void insert(int value){
        this.tree = insert(tree,value);
    }

    private AvlNode insert(AvlNode node,int value){
        AvlNode p;
        if (node == null){
            node = createNode(value,0);
        }else {
            if (value > node.value){
                node.right = insert(node.right, value);
                if (Math.abs(hight(node.right) - hight(node.left)) > 1) {//失衡类型为 RR or RL
                    p = node;
                    while (p.right != null){
                        p = p.right;
                    }
                    if (value == p.value) {//说明是RR
                        node = rrRotaing(node);
                    } else {
                        node = rlRotaing(node);
                    }
                }
            }else if (value < node.value){
                node.left = insert(node.left,  value);
                if (Math.abs(hight(node.right) - hight(node.left)) > 1) {//失衡类型为 LL or LR
                    p = node;
                    while (p.left != null){
                        p = p.left;
                    }
                    if (value == p.value) {//说明是LL
                        node = llRotaing(node);
                    } else {//否则为LR
                        node = lrRotaing(node);
                    }
                }
            }
        }
        node.hight = getMaxHeight(hight(node.left), hight(node.right)) + 1;
        return node;
    }


    /**
     * LL
     *
     * @param node 失衡节点
     * @return 左旋后的根节点
     */
    public AvlNode llRotaing(AvlNode node) {
        AvlNode avl_lift = node.left;//失衡节点的左子树节点 旋转后的根节点
        //把失衡节点左子树的右子树节点挂到失衡节点的左子树上
        node.left = avl_lift.right;
        //把失衡节点作为该节点左子树的右子树
        avl_lift.right = node;
        //重新计算失衡节点和失衡节点左子树(即旋转后的根节点)旋转后的节点高度
        node.hight = getMaxHeight(hight(node.left), hight(node.right)) + 1;
        avl_lift.hight = getMaxHeight(hight(avl_lift.left), hight(avl_lift.right)) + 1;//得到左子树和右子树中最大的高度 + 1
        return avl_lift;
    }

    /**
     * RR
     *
     * @param node 失衡节点
     * @return 右旋后的根节点
     */
    public AvlNode rrRotaing(AvlNode node) {
        AvlNode avl_right = node.right;//失衡节点的右子树节点 旋转后的根节点
        //把失衡节点右子树的左子树节点挂到失衡节点的右子树上
        node.right = avl_right.left;
        //把失衡节点作为该节点右子树的左子树
        avl_right.left = node;
        //重新计算失衡节点和失衡节点右子树(即旋转后的根节点)旋转后的节点高度
        node.hight = getMaxHeight(hight(node.left), hight(node.right)) + 1;
        avl_right.hight = getMaxHeight(hight(avl_right.left), hight(avl_right.right)) + 1;//得到左子树和右子树中最大的高度 + 1
        return avl_right;
    }

    /**
     * LR
     * 1. 先把失衡节点的左子树做一次rr旋转
     * 2. 把该数做一次ll旋转
     *
     * @param node 失衡节点
     * @return
     */
    public AvlNode lrRotaing(AvlNode node) {
        node.left = rrRotaing(node.left);
        AvlNode avlNode = llRotaing(node);
        return avlNode;
    }

    /**
     * RL
     * 1. 先把失衡节点的右子树做一次ll旋转
     * 2. 把该数做一次rr旋转
     *
     * @param node 失衡节点
     * @return
     */
    public AvlNode rlRotaing(AvlNode node) {
        node.right = llRotaing(node.right);
        AvlNode avlNode = rrRotaing(node);
        return avlNode;
    }

    //获取节点高度
    public int hight(AvlNode node) {
        return node == null ? 0 : node.hight;
    }

    //获取树的高度
    public int hight() {
        return hight(tree);
    }

    //获取高度最大值
    private int getMaxHeight(int height1, int height2) {
        return height1 > height2 ? height1 : height2;
    }

    /**
     * 中序遍历
     * 先打印左子树
     * 然后打印自身
     * 最后打印右子树
     *
     * @param node
     */
    public void inOrder(AvlNode node) {
        if (node == null) {
            return;
        }
        inOrder(node.left);
        System.out.print(node.getValue() + "->");
        inOrder(node.right);
    }

    /**
     * 前序遍历
     * 先打印自身
     * 然后打印左子树
     * 最后打印右子树
     *
     * @param node
     */
    public void preOrder(AvlNode node) {
        if (node == null) {
            return;
        }
        System.out.print(node.getValue() + "->");
        preOrder(node.left);
        preOrder(node.right);
    }

    /**
     * 后续遍历
     * 先打印左子树
     * 然后打印右子树
     * 最后打印自身
     *
     * @param node
     */
    public void postOrder(AvlNode node) {
        if (node == null) {
            return;
        }
        postOrder(node.left);
        postOrder(node.right);
        System.out.print(node.getValue() + "->");
    }

    public static AvlNode createNode(int value, AvlNode left, AvlNode right, int hight) {
        return new AvlNode(value, left, right, hight);
    }

    public static AvlNode createNode(int value, int hight) {
        return new AvlNode(value, null, null, hight);
    }

    public static AvlNode createNode(int value, AvlNode left, AvlNode right) {
        return new AvlNode(value, left, right);
    }

    public static AvlNode createNode(int value) {
        return new AvlNode(value, null, null);
    }

    //avl树节点类
    private static class AvlNode {
        private int value;
        private AvlNode left;
        private AvlNode right;
        private int hight;//定义节点高度

        public AvlNode(int value, AvlNode left, AvlNode right, int hight) {
            this.value = value;
            this.left = left;
            this.right = right;
            this.hight = hight;
        }

        public AvlNode(int value, AvlNode left, AvlNode right) {
            this(value, left, right, 0);
        }

        public int getValue() {
            return value;
        }

        public void setValue(int value) {
            this.value = value;
        }

        public AvlNode getLeft() {
            return left;
        }

        public void setLeft(AvlNode left) {
            this.left = left;
        }

        public AvlNode getRight() {
            return right;
        }

        public void setRight(AvlNode right) {
            this.right = right;
        }

        public int getHight() {
            return hight;
        }

        public void setHight(int hight) {
            this.hight = hight;
        }
    }
}

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值