Leetcode450
链接:力扣 。
题目:
给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
一般来说,删除节点可分为两个步骤:
首先找到需要删除的节点; 如果找到了,删除它。 说明: 要求算法时间复杂度为 $O(h)$,h 为树的高度。
示例:
输入: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],只要删除后仍然是BST即可。
思路:
要删除一个结点,首先要查找该结点。对于查找结点的方法,已经在前面的题目中出现(Leetcode700)。
根据查找结点的结果,分为以下几类情况:
- 第一种情况:没找到删除的节点,遍历到空节点直接返回了
- 找到删除的节点
- 第二种情况:左右孩子都为空(叶子节点),直接删除节点, 返回NULL为根节点
- 第三种情况:删除节点的左孩子为空,右孩子不为空,删除节点,右孩子补位,返回右孩子为根节点
- 第四种情况:删除节点的右孩子为空,左孩子不为空,删除节点,左孩子补位,返回左孩子为根节点
- 第五种情况:左右孩子节点都不为空,则将删除节点的左子树头结点(左孩子)放到删除节点的右子树的最左面节点的左孩子上,返回删除节点右孩子为新的根节点。
综上,我们就可以得到本题的完整思路了。
从根结点开始,查找当前结点root是否为目标Key结点。如果root的值大于Key,则去其左子树查找;如果root的值小于Key,则去其右子树查找;如果root为空,说明查找失败返回空值;如果root的值等于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 == nullptr) {
return nullptr;
}
if (root->val == key) {
if (root->left == nullptr && root->right == nullptr) {
delete root;
return nullptr;
}
else if (root->left != nullptr && root->right == nullptr) {
TreeNode *node = root->left;
delete root;
return node;
}
else if (root->left == nullptr && root->right != nullptr) {
TreeNode *node = root->right;
delete root;
return node;
}
else {
TreeNode *node = root->right;
while (node->left) {
node = node->left;
}
node->left = root->left;
node = root->right;
delete root;
return node;
}
}
else if (root->val > key) {
root->left = deleteNode(root->left, key);
return root;
}
else {
root->right = deleteNode(root->right, key);
return root;
}
}
};