450. 删除二叉搜索树中的节点

2023.6.16
首先考虑找到待删除节点后应当怎么办,其实就是两种情况:
1.待删除节点的右节点不为None,则将其左节点接到右节点的最左侧,然后将右节点连到删除节点的父节点即可。
2.待删除节点的右节点不存在为None,则直接将左节点连到删除节点的父节点即可。
本来我的考虑是如果要删除当前节点,应当要拿到当前节点的父节点才行,但实际上只能说递归真的很神奇。

首先是递归的返还条件是遇到空节点
之后是递归的逻辑,如果当前节点是待删除节点,返还删除当前节点后的新节点,如果当前节点不是待删除节点,遍历当前节点的左节点或右节点,同时返还值是当前节点。

class Solution:
    def deleteNode(self, root: Optional[TreeNode], key: int) -> Optional[TreeNode]:
        if not root:
            return root

        # 中
        if root.val == key:
            if root.right == None:  # 右面是空的
                return root.left
        
            elif root.left == None:
                return root.right
            
            else:
                keeper = right = root.right
                while right.left:
                    right = right.left
                right.left = root.left
                return keeper
        
        if root.val > key:
            root.left = self.deleteNode(root.left, key)
        
        else: root.right = self.deleteNode(root.right, key)

        return root
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值