题目
给定一个二叉搜索树的根节点 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 为树的高度。
解答
直接上代码：
/**
* 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 == nullptr) {
return nullptr;
}
if (root->val == key) {
// 当前节点需要删除
if (root->left == nullptr) {
// 没有左子树，返回右子树
return root->right;
} else if (root->right == nullptr) {
// 没有右子树，返回左子树
return root->left;
} else {
// 左右子树都存在，找到右子树中的最小节点
TreeNode* minNode = findMin(root->right);
// 将最小节点的值赋给当前节点
root->val = minNode->val;
// 在右子树中递归删除最小节点
root->right = deleteNode(root->right, minNode->val);
}
} else if (root->val > key) {
// 要删除的值小于当前节点的值，递归在左子树中删除
root->left = deleteNode(root->left, key);
} else {
// 要删除的值大于当前节点的值，递归在右子树中删除
root->right = deleteNode(root->right, key);
}
return root;
}
private:
// 在二叉搜索树中找到最小节点
TreeNode* findMin(TreeNode* node) {
while (node->left != nullptr) {
node = node->left;
}
return node;
}
};