AVL树和他的旋转

AVL树


  AVL树分其节点有:父亲节点,val值,左孩子节点,右孩子节点,还有平衡因子。在这里的平衡因子属于int类型。


  AVL树的插入规则,就是左子树的val值要始终小于右子树的val值,并且其平衡因子的绝对值不能大于2,如果平衡因子大于2需要进行调节。

  当前节点的平衡因子=右子树高度-左子树的高度。但是,不是每棵树,都必须有平衡因子,这只是其中的一种实现 方式。

一棵AVL树或者是空树,或者是具有以下性质的二叉搜索树:

1. 它的左右子树都是AVL树

2. 左右子树高度之差(简称平衡因子)的绝对值不超过1(-1/0/1) 如果一棵二叉搜索树是高度平衡的,它就是AVL树。

如图:

如果一棵二叉搜索树是高度平衡的,它就是AVL树。如果它有n个结点,其高度可保持在 O(log2n),搜索时间复杂 度O(log2n )。


AVL树的插入和旋转


  AVL树就是在二叉搜索树的基础上引入了平衡因子,因此AVL树也可以看成是二叉搜索树。那么AVL树的插入过程可 以分为两步:

1. 按照二叉搜索树的方式插入新节点

2. 调整节点的平衡因子


在这里最重要的就是旋转了,这里我以右旋为例子。

如图:

此时在这棵子树上的右边插入,值为60的这个节点的平衡因子的绝对值大于1了,此时我们需要对其进行右旋,调节平衡因子。

这里我们应该怎么旋转呢?

我们只需要把30的这个节点,向上一提就可以了。

如图:

 上面就是调整之后的样子。


那代码应该来怎么来写呢,这里我会先用图来展示。

如图:

 这里的箭头代表了节点的指向,而这里的30节点多出来的箭头指向为60节点之前的父母,需要判断一下60是左孩子还是右孩子。英文字母表示了重新定义出来的节点用来表示。

如代码:

public class AVL {
    static class TreeNode{
        public int val;
        public int bf;//平衡因子
        public TreeNode right;//右孩子的引用
        public TreeNode left;//左孩子的引用
        public TreeNode parent;//父亲节点
        public TreeNode(int val){
            this.val = val;
        }
    }
    public TreeNode root;//根节点
    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(cur.val < val){
                parent = cur;
                cur = cur.right;
            }
            else if (cur.val == val) {
                return false;
            }
            else{
                parent = cur;
                cur = cur.left;
            }
        }
        if(parent.val < val){
            parent.right = node;
        }
        else{
            parent.left = node;
        }
        node.parent = parent;
        cur.parent = node;
        while(parent != null){
            if(cur == parent.right){
                parent.bf++;
            }
            else{
                parent.bf--;
            }
            if(parent.bf == 0){
                break;
            } else if (parent.bf == 1 || parent.bf == -1) {
                cur = parent;
                parent = cur.parent;
            }
            else{
                if(parent.bf == 2){
                    if(cur.bf == 1){

                    }
                    else{

                    }
                }
                else{
                    if(cur.bf == -1){
                        rotateRight(parent);
                    }
                    else{

                    }

                }
                break;
            }
        }
        return false;
    }
    private void rotateRight(TreeNode parent){
        TreeNode subl = parent.left;
        TreeNode sublR = subl.right;
        parent.left = sublR;
        subl.right = parent;
        if(sublR != null) {
            sublR.parent = parent;
        }
        TreeNode pParent = parent.parent;
        if(parent == root){
            root = subl;
            root.parent = null;
        }
        else {
            if(pParent.right == parent){
                pParent.right = subl ;
            }
            else {
                pParent.left = subl;
            }
        }
        subl.bf = 0;
        parent.bf = 0;
    }


}

这里的代码我只完成了右旋,左旋类似,还有左右双旋,感兴趣的可以挑战一下。

  • 4
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值