本文用python3实现二叉查找树的查找、插入、删除操作。
完整代码:https://github.com/toanoyx/BasicAlgorithm/tree/master/tree
查找
查找操作:先取根节点,如果它等于要查找的数那就返回。如果要查找的数比根节点的值小,就在左子树中递归查找;如果要查找的数比根节点的值大,那就在右子树中递归查找。
def search(self, data):
res = []
node = self.root
while node:
if data < node.val:
node = node.left
else:
if data == node.val:
res.append(node)
node = node.right
return res
插入
插入操作:从根节点开始,依次比较要插入的数据与节点的大小关系。如果插入的数据比节点的大,并且节点的右子树为空,就将数据直接插到右子节点的位置;如果右子树不为空,就在递归遍历右子树,查找插入位置。如果要插入的数据比节点的小,并且节点的左子树为空,就将数据直接插到左子树的位置;如果左子树不为空,就再递归遍历左子树,查找插入位置。