平衡二叉搜索树(AVL树)的左旋、右旋(python)

AVL树是可以自平衡的二叉搜索树(参考:二叉搜索(排序)树的python实现

AVL树的性质:

1、树的每个节点的左右子树的高度差的绝对值小于等于1

2、AVL树是一种特殊的二叉搜索树

当插入一个节点时候,可能会导致AVL树的失衡,此时可以通过左旋、右旋、先左旋再右旋和先右旋再左旋的4种方法保持树的平衡

​
class AVL_node(Node):
    def __init__(self,val):
        Node.__init__(self,val)
        self.bf=0 #平衡因子

class AVL(BST):
    def __init__(self,node_list):
        BST.__init__(self,node_list)

​

左旋

    def rotate_left(self,a,b): #左旋
        a.rt=b.lt 
        if b.lt:
            b.lt.parent=a
        a.parent=b
        b.lt=a
        a.bf=0  #更新平衡因子
        b.bf=0

右旋

    def rotate_right(self,a,b): #右旋
        if b.rt:
            b.rt.parent=a
        a.lt=b.rt
        b.rt=a
        a.parent=b
        a.bf=0 #更新平衡因子
        b.bf=0

先左旋再右旋

    def rotate_lt_rt(self,a,b):
        c=b.rt  #左旋
        if c.lt:  
            c.lt.parent=b
        b.rt=c.lt
        c.lt=b
        b.parent=c  
        if c.rt: #右旋
            c.rt.parent=a
        a.lt=c.rt
        a.parent=c
        c.rt=a
        if c.bf<0:  #更新平衡因子
            a.bf=0
            b.bf=1
        else:
            a.bf=-1
            b.bf=0  

先右旋再左旋

    def rotate_rt_lt(self,a,b): #右旋 左旋
        c=b.lt #右旋
        if c.rt:
            c.rt.parent=b
        b.lt=c.rt
        b.parent=c
        c.rt=b  
        if c.lt: #左旋
            c.lt.parent=a
        a.rt=c.lt
        a.parent.c
        c.lt=a 
        if c.bf>0: #更新平衡因子
            a.bf=0
            b.bf=-1
        else:
            a.bf=1
            b.bf=0
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值