思路:
一、key不在二叉搜索树(BST)中,则依此确定递归停止条件,返回None;
二、key在BST:
1.左右子孩子为None,直接删了完事儿,返回None,与第2.情况一致,可以共用代码;
2.左孩子为None,则返回右孩子为根节点;
3.右孩子为None,则返回左孩子为根节点;
4.左孩子L和右孩子R都不为None,则找到右子树R的最底层的最左孩子UL。然后让L成为UL的左孩子。
注意:
当key不在BST中时,则遍历BST的某一分支都找不到,对吧?那么从这一分支的最底层返回None开始,逐层按原顺序返回。即返回到第一层,此时的root和一开始传入的待查找root是一模一样的BST。
递归法:
# 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]:
if not root:
return None
if root.val == key:
if not root.left: return root.right
elif not root.right: return root.left
else:
cur = root.right
while cur.left:
cur = cur.left
cur.left = root.left
return root.right
if key < root.val: root.left = self.deleteNode(root.left, key)
if key > root.val: root.right = self.deleteNode(root.right, key)
return root
总结:
读完本篇,大家会发现二叉搜索树删除节点比增加节点复杂的多。
因为二叉搜索树添加节点只需要在叶子上添加就可以的,不涉及到结构的调整,而删除节点操作涉及到结构的调整。
这里我们依然使用递归函数的返回值来完成把节点从二叉树中移除的操作。
这里最关键的逻辑就是第五种情况(删除一个左右孩子都不为空的节点),这种情况一定要想清楚。
而且就算想清楚了,对应的代码也未必可以写出来,所以这道题目既考察思维逻辑,也考察代码能力。
References:
leetcode-master/0450.删除二叉搜索树中的节点.md at master · youngyangyang04/leetcode-master · GitHub