题目描述:
给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。
一般来说,删除节点可分为两个步骤:
- 首先找到需要删除的节点;
- 如果找到了,删除它。
算法思想:
根据二叉搜索树的性质
- 如果目标节点大于当前节点值,则去右子树中删除;
- 如果目标节点小于当前节点值,则去左子树中删除;
- 如果目标节点就是当前节点,分为以下三种情况:
- 其无左子:其右子顶替其位置,删除了该节点;
- 其无右子:其左子顶替其位置,删除了该节点;
- 其左右子节点都有:其左子树转移到其右子树的最左节点的左子树上,然后右子树顶替其位置,由此删除了该节点。
第三种情况图示如下:
代码:
TreeNode* deleteNode(TreeNode* root, int key) {
if(root==NULL)return NULL;
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) return root->right;//左子树空,右子树的根结点
if(!root->right)return root->left;//右子树空,左子树的根节点
TreeNode*cur=root->right;//从右树中寻找最左边的的结点
while(cur->left)
cur=cur->left;
cur->left=root->left;
root=root->right;
}
return root;
}