leetcode 450. 删除二叉搜索树中的节点 medium
题目描述:
给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
一般来说,删除节点可分为两个步骤:
首先找到需要删除的节点;
如果找到了,删除它。
说明: 要求算法时间复杂度为 O(h),h 为树的高度。
解题思路:
其实做这题最困惑的是,为什么传指针也能达到删除的目的? 后来想明白了,如果参数是指针的引用,你把这个参数修改就能起到删除的目的,并不需要返回什么东西,但是如果传的是指针,你需要返回root指针,并且每一步的递归调用,你都要用->把对应的值给修改了(比如root->right=deleteNode(root->right,key))
1)假如要删除的节点,两孩子都空 or 左空 右不空 or左不空,右空 也就是至少一个孩子是空
是一个逻辑,让root= 空对面的那个孩子(可能不空,也可能都空)
2) 假如 要删除的节点左右都不空,那就去把它的后继节点(右子树的最左节点的值给拿过来,再去把那个右子树的最左节点给删了)
由于BST的左<根<右的性质,使得我们可以快速定位到要删除的节点,我们对于当前节点值不等于key的情况,根据大小关系对其左右子节点分别调用递归函数。若当前节点就是要删除的节点,我们首先判断是否有一个子节点不存在,那么我们就将root指向另一个节点,如果左右子节点都不存在,那么root就赋值为空了,也正确。难点就在于处理左右子节点都存在的情况,我们需要在右子树找到最小值,即右子树中最左下方的节点,然后将该最小值赋值给root,然后再在右子树中调用递归函数来删除这个值最小的节点。
代码:
迭代版
/**
* 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 root;
TreeNode *pp = nullptr, *p = root;
while (p && p->val != key){
pp = p;
if (key > p->val)
p = p->right;
else
p = p->left;
}
if (!p)
return root;
if (p->left && p->right){
TreeNode *min_pp = p;
TreeNode *min_p = p->right;
while (min_p->left){
min_pp = min_p;
min_p = min_p->left;
}
p->val = min_p->val;
pp = min_pp;
p = min_p;
}
TreeNode *children = p->left?p->left:p->right;
if (pp == nullptr)
root = children;
else if (pp->left == p)
pp->left = children;
else
pp->right = children;
delete p;
return root;
}
};
递归版
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* deleteNode(TreeNode* root, int key) {
if( root==nullptr)
return nullptr;
if(key>root->val)
root->right=deleteNode(root->right,key); // 这一步很关键,理解为什么传指针,也能做到删除,因为全程都在用->!!!
else if(key<root->val)
root->left=deleteNode(root->left,key);
else{
if( !root->left || !root->right){
TreeNode* old=root;
root=(root->left)?root->left:root->right;
delete old;
} // 两个孩子至少有一个没了
else{
TreeNode *cur=root->right;
while(cur->left)
cur=cur->left;
root->val=cur->val;
root->right=deleteNode(root->right,root->val); //这里也是 root-》right来设置!!!
}
}
return root;
}
};