给定一个二叉搜索树的根节点 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 输出: []
思路:二叉搜索树的删除操作,比插入操作麻烦很多,要考虑的情况有很多。
情况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