450. 删除二叉搜索树中的节点
LeetCode题目要求
给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
一般来说,删除节点可分为两个步骤:
首先找到需要删除的节点;
如果找到了,删除它。
![图](/i/l/?n=23&i=blog/458364/202301/458364-20230125214609906-2050949719.png)
示例
输入: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]。
解题思路
由于是删除节点操作,删除后还需要保证是一个二叉搜索树。
- 如果删除的是叶子节点,那么不需要又任何变化,直接删除即可
- 如果是非叶子节点,存在左右子树,将右子树升级为父结点
- 如果是非叶子节点,只存在左子树,那么左子树升级为父结点
- 如果是非叶子节点,只存在右子树,那么右子树升级为父节点
上代码
class Solution {
public TreeNode deleteNode(TreeNode root, int key) {
/**
1. 如果删除的是叶子节点,那么不需要又任何变化,直接删除即可
2. 如果是非叶子节点,存在左右子树,将右子树升级为父结点
3. 如果是非叶子节点,只存在左子树,那么左子树升级为父结点
4. 如果是非叶子节点,只存在右子树,那么右子树升级为父节点
*/
if (root == null) {
return root;
}
if (root.val == key) {
if (root.left == null && root.right == null) {
return null;
} else if (root.left == null) {
return root.right;
} else if (root.right == null) {
return root.left;
} else {
TreeNode node = root.right;
while (node.left != null) {
node = node.left;
}
node.left = root.left;
root = root.right;
return root;
}
}
// 往左子树遍历
if (root.val > key) {
root.left = deleteNode(root.left, key);
}
// 往右子树遍历
if (root.val < key) {
root.right = deleteNode(root.right, key);
}
return root;
}
}
重难点
搞明白以下集中情况, 并结合二叉搜索树的特点
- 如果删除的是叶子节点,那么不需要又任何变化,直接删除即可
- 如果是非叶子节点,存在左右子树,将右子树升级为父结点
- 如果是非叶子节点,只存在左子树,那么左子树升级为父结点
- 如果是非叶子节点,只存在右子树,那么右子树升级为父节点
附:学习资料链接