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

给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。

一般来说,删除节点可分为两个步骤:

  1. 首先找到需要删除的节点;
  2. 如果找到了,删除它。

示例 1:

输入:root = [5,3,6,2,4,null,7], key = 3
输出:[5,4,6,2,null,null,7]
解释:给定需要删除的节点值是 3,所以我们首先找到 3 这个节点,然后删除它。
一个正确的答案是 [5,4,6,2,null,null,7], 如下图所示。
另一个正确答案是 [5,2,6,null,4,null,7]。


示例 2:

输入: root = [5,3,6,2,4,null,7], key = 0
输出: [5,3,6,2,4,null,7]
解释: 二叉树不包含值为 0 的节点

示例 3:

输入: root = [], key = 0
输出: []

思路:二叉搜索树的删除操作,比插入操作麻烦很多,要考虑的情况有很多。

情况1:若树为空,则不产生删除,直接返回None

情况2:删除的结点是个叶节点,不对其他节点产生影响,直接删除返回None即可

情况3:删除的结点不是叶节点,但只有右孩子没有左孩子,则用右孩子补上这个结点

情况4:删除的结点不是叶节点,但只有左孩子没有右孩子,则用左孩子补上这个结点

情况5:删除的结点既有左孩子,又有右孩子,这种情况就比较麻烦了,因为删除过后还要保证整棵树是二叉搜索树,你得保证有序性,因此这种情况下,需要把删除的结点的左孩子挂在删除的结点的右孩子的最左边的那个节点的左孩子上,这样就能保证有序,因为要删除的这个结点的左孩子一定小于要删除的这个结点的右孩子中最小的一个结点,也就是右孩子中最左边的那个结点。

代码(Python):

class Solution(object):
    def deleteNode(self, root, key):
        if not root:                          #情况1
            return None
        if root.val > key:
            root.left = self.deleteNode(root.left,key)
        if root.val < key:
            root.right = self.deleteNode(root.right,key)
        if root.val == key:
            if not root.left and not root.right:    #情况2
                return None
            elif not root.left:                     #情况3
                return root.right
            elif not root.right:                    #情况4
                return root.left
            else:                                   #情况5
                cur = root.right
                while cur.left:
                    cur = cur.left
                cur.left = root.left
                return root.right
        return root
  • 5
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值