来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/delete-node-in-a-bst
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
450.给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
一般来说,删除节点可分为两个步骤:
首先找到需要删除的节点;
如果找到了,删除它。
说明: 要求算法时间复杂度为 O(h),h 为树的高度。
- 删除二叉搜索的某个结点,可以通过递归来完成。在递归的过程需要考虑五种情况:(1)没找到,直接返回NULL;(2)找到要删除的结点并且左右孩子都为空,则删除后返回NULL;(3)找到要删除的结点且左孩子空右孩子不空,则删除结点后返回右孩子;(4)找到要删除的结点且左孩子不为空右孩子空,则删除结点后返回左孩子;(5)找到要删除的结点且左右孩子都不为空,则将左孩子接到右孩子最左侧后再返回右孩子。这里前四步都很容易想到,比较难想到的是第五步。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
struct TreeNode* deleteNode(struct TreeNode* root, int key){
if (root == NULL) //第一种情况,没有找到要删除的结点
return root;
if (root->val == key) {
if (root->left == NULL && root->right == NULL) { //第二种情况,找到要删除的结点,且该结点左右结点都为空,则删除结点然后返回NULL即可
free(root);
return NULL;
} else if (root->left == NULL) { //第三种情况,找到要删除的结点,且该结点左孩子为空右孩子不为空,则删除结点并返回该结点的右孩子
struct TreeNode* p = root->right;
free(root);
return p;
} else if (root->right == NULL) { //第四种情况,找到要删除的结点,且该结点左孩子不为空,右孩子为空,则删除结点并返回该结点的左孩子
struct TreeNode* p = root->left;
free(root);
return p;
} else { //第五种情况,找到要删除的结点,且该结点左右孩子都不为空,则删除结点,并把该结点的左孩子接到右孩子的最左侧,然后返回该结点的右孩子
struct TreeNode* p = root->right;
while (p->left != NULL) {
p = p->left;
}
p->left = root->left;
p = root->right;
free(root);
return p;
}
}
else if (root->val > key)
root->left = deleteNode(root->left, key);
else
root->right = deleteNode(root->right, key);
return root;
}
- 这个题用迭代法的话并没有递归写起来这么简单,在删除结点后将剩下部分与父节点连接就有很多种要判断的情况。要判断要删除的结点的父节点是否为空、不为空的话要判断要删除的结点是父节点的左孩子还是父节点的右孩子。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
struct TreeNode* deleteOneNode(struct TreeNode* target) {
if (target->left == NULL && target->right == NULL)
return NULL;
if (target->right == NULL)
return target->left;
if (target->left == NULL)
return target->right;
struct TreeNode* p = target->right;
while (p->left) {
p = p->left;
}
p->left = target->left;
return target->right;
}
struct TreeNode* deleteNode(struct TreeNode* root, int key){
if (root == NULL)
return root;
struct TreeNode* p = root;
struct TreeNode* pre = NULL;
while (p) {
if (p->val == key)
break;
pre = p;
if (p->val > key)
p = p->left;
else
p = p->right;
}
if (pre == NULL)
return deleteOneNode(p);
if (pre->left && pre->left->val == key)
pre->left = deleteOneNode(p);
if (pre->right && pre->right->val == key)
pre->right = deleteOneNode(p);
return root;
}