题目描述
给定一个二叉搜索树的根节点 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
解题思路
-
基本情况:
- 如果
root
为nullptr
,直接返回nullptr
,表示树为空或已达到叶节点的子节点。
- 如果
-
找到目标节点:
- 如果
root->val == key
,则找到了需要删除的节点。根据该节点的子节点情况处理:- 两个子节点:找到左子树的最右节点(最大值)或右子树的最左节点(最小值),用其值替代当前节点值,并递归删除该替代节点。
- 一个子节点或无子节点:将节点替换为其非空的子节点,如果两个子节点都为空,则将其设置为
nullptr
。
- 如果
-
递归搜索并删除不匹配节点:
- 如果
root->val > key
,递归到左子树寻找并尝试删除。 - 如果
root->val < key
,递归到右子树寻找并
- 如果
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
TreeNode* deleteNode(TreeNode* root, int key) {
if (!root) {
return nullptr;
}
if (root->val == key) {
if (root->left) {
auto lmp = root->left;
while (lmp->right) {
lmp = lmp->right;
}
int lm = lmp->val;
root->val = lm;
root->left = deleteNode(root->left, lm);
} else if (root->right) {
auto rmp = root->right;
while (rmp->left) {
rmp = rmp->left;
}
int rm = rmp->val;
root->val = rm;
root->right = deleteNode(root->right, rm);
} else {
root = nullptr;
}
} else if (root->val > key) {
root->left = deleteNode(root->left, key);
} else {
root->right = deleteNode(root->right, key);
}
return root;
}
};