python二分查找树(BST)和AVL树

本文介绍了Python中二分查找树(BST)的基本概念和实现,包括其数据结构特性,有助于快速定位数据。同时,文章探讨了AVL树作为BST的改进,通过平衡因子确保左右子树平衡,从而保持稳定的查询效率。内容包含BST和AVL树的Python代码实现。
摘要由CSDN通过智能技术生成

BST树

二分查找树是左子节点小于根节点,根节点小于右子节点的树形数据结构,有利于数据的快速定位,该python实现,含有两个类,TreeNode和BinarySearchTree,下面逐行注释代码方便理解

class BinarySearchTree:
    def __init__(self):
        # 定义根节点
        self.root = None
        # 用于记录树的大小,即树中有多少个node
        self.size = 0
    
    def put(self,key,val):
    	# 新node加入树中
        if self.root:
            self._put(key,val,self.root)
        else:
            self.root = TreeNode(key,val)
        self.size = self.size + 1

    def _put(self,key,val,currentNode):
    	# 如果key小于根节点key则插入左子树,如果key大于根节点key则插入右子树
        if key < currentNode.key:
            if currentNode.hasLeftChild():
                self._put(key,val,currentNode.leftChild)
            else:
                currentNode.leftChild = TreeNode(key,val,parent=currentNode)
        else:
            if currentNode.hasRightChild():
                self._put(key,val,currentNode.rightChild)
            else:
                currentNode.rightChild = TreeNode(key,val,parent=currentNode)
            
    def __setitem__(self,k,v):
        self.put(k,v)

    def get(self,key):
    	# 通过key过去bst树的val
        if self.root:
            res = self._get(key,self.root)
            if res:
                return res.payload
            else:
                return None
        else:
            return None
        
    def _get(self,key,currentNode):
        if not currentNode:
            return None
        elif currentNode.key == key:
            return currentNode
        elif key < currentNode.key:
            return self._get(key,currentNode.leftChild)
        else:
            return self._get(key,currentNode.rightChild)
            
        
    def __getitem__(self,key):
    	# 用于类似map的访问比如a['key']=
        res = self.get(key)
        if res:
            return res
        else:
            raise KeyError('Error, key not in tree')
            

    def __contains__(self,key):
    	# 用于判断是否含有key
        if self._get(key,self.root):
            return True
        else:
            return False
        
    def length(self):
        return self.size

    def __len__(self):
        return self.size

    def __iter__(self):
        return self.root.__iter__()
    
    # 删除bst树中的节点
    def delete(self,key):
    	
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值