二叉排序树的插入和删除操作(python实现)

二叉排序树的插入和删除操作都是在保持二叉排序树特性的前提下进行的。

插入操作:

在二叉排序树中插入一个新节点时,先比较新节点的值和当前节点的值的大小关系,若小于当前节点,则继续在当前节点的左子树中查找;若大于当前节点,则继续在当前节点的右子树中查找。重复该过程,直到找到一个空节点,将新节点插入到该位置上。

删除操作:

删除操作比较复杂,需要分类讨论。

若待删除节点为叶子节点,直接删除即可。
若待删除节点只有一个子节点,将其子节点替代自身即可。
若待删除节点有两个子节点,则需要寻找其右子树中的最小值节点或左子树中的最大值节点来替代自身,然后再将该最小值节点或最大值节点删除。

具体过程如下:

  • 在二叉排序树中查找待删除节点,并记录其父节点;
  • 对待删除节点进行分类讨论,若为叶子节点或只有一个子节点,则直接删除;
  • 若待删除节点有两个子节点,则找到其右子树中的最小值节点,或左子树中的最大值节点;
  • 将该最小值节点或最大值节点替代待删除节点,并将其子节点链接到其父节点上; 若待删除节点为根节点,则替换根节点。
  • 需要注意的是,在删除节点之后,为了保持二叉排序树的特性,需要对该节点的父节点及其祖先节点进行平衡调整。
class Node:
    def __init__(self, key):
        self.left = None
        self.right = None
        self.key = key

def insert(root, key):
    if root is None:
        return Node(key)
    if key < root.key:
        root.left = insert(root.left, key)
    elif key > root.key:
        root.right = insert(root.right, key)
    return root

def minValueNode(node):
    current = node
    while(current.left is not None):
        current = current.left
    return current

def deleteNode(root, key):
    if root is None:
        return root
    if key < root.key:
        root.left = deleteNode(root.left, key)
    elif(key > root.key):
        root.right = deleteNode(root.right, key)
    else:
        if root.left is None:
            temp = root.right
            root = None
            return temp
        elif root.right is None:
            temp = root.left
            root = None
            return temp
        temp = minValueNode(root.right)
        root.key = temp.key
        root.right = deleteNode(root.right, temp.key)
    return root

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Tomorrowave

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值