题目地址(450. 删除二叉搜索树中的节点)

题目地址(450. 删除二叉搜索树中的节点)

https://leetcode.cn/problems/delete-node-in-a-bst/

题目描述

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

一般来说，删除节点可分为两个步骤：

首先找到需要删除的节点；
如果找到了，删除它。

 

示例 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
输出: []

 

提示:

节点数的范围 [0, 104].
-105 <= Node.val <= 105
节点值唯一
root 是合法的二叉搜索树
-105 <= key <= 105

 

进阶： 要求算法时间复杂度为 O(h)，h 为树的高度。

前置知识

公司

• 暂无

思路

关键点

代码

• 语言支持：Python3

Python3 Code:

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def deleteNode(self, root: Optional[TreeNode], key: int) -> Optional[TreeNode]:
        cur,curParent = root,None
        while cur and cur.val != key:
            curParent = cur
            cur = cur.left if cur.val > key else cur.right
        if cur is None:
            return root # 如果找不到 返回原来的树
        # 如果找到叶子节点 
        if cur.left is None and not cur.right:
            cur = None
        elif cur.right is None: # 没有右子树
            cur = cur.left
        elif cur.left is None: # 没有左子树
            cur = cur.right
        else:# 既有左子树又有右子树
            successor,successorParent = cur.right,cur
            while successor.left:# 寻找第一个大于cur的数
                successorParent = successor
                successor = successor.left
            if successorParent.val == cur.val:# 右边没有左子树，右边最小的是根
                successorParent.right = successor.right
            else:# successor表示找到的第一个比key大的值 如果有左子树 那么将右子树合并到successorParent的做左边
                successorParent.left = successor.right 
            successor.right = cur.right
            successor.left = cur.left
            cur = successor
        if curParent is None: #找不到
            return cur 
        if curParent.left and curParent.left.val == key:
            curParent.left = cur 
        else:
            curParent.right = cur 
        return root
        

复杂度分析

令 n 为数组长度。

• 时间复杂度：$O(n)$
• 空间复杂度：$O(1)$

Reference:

https://leetcode.cn/problems/delete-node-in-a-bst/solution/shan-chu-er-cha-sou-suo-shu-zhong-de-jie-n6vo/