非最优解,仅学习记录,参考了题解的思路。
算法思路:
情况大概几种:
- 根节点为空,则直接返回根节点
- 根节点不空,正好是要删除的结点,判断左右子树是否为空
- 都为空则直接删除
- 一边不空则子树直接变根结点
- 两边都不空则左子树去右子树最左结点下
- 根节点不空,小于要删除的结点,则查找右子树,递归直到情况2
- 根节点不空,大于要删除的结点,则查找左子树,递归直到情况2
注意:
- 根节点的引用指即返回根节点
- 二叉树的删除结点需释放空间即delete 结点
- 二叉搜索树即二叉排序树,拥有所有左子树均大于右子树的性质。
*/
class Solution {
public:
TreeNode* deleteNode(TreeNode* root, int key) {
//情况1:根节点为空,则直接返回根节点
if(root==nullptr)return root;
//情况2:根节点不空,正好是要删除的结点,判断左右子树是否为空,
if(root->val==key){
TreeNode* cur;
if(root->left==nullptr && root->right==nullptr){
//左右子树都空则直接删除,返回空
delete root;
return nullptr;
}
else if(root->left==nullptr){
//左子树空返回右子树,并删除结点
cur=root->right;
delete root;
return cur;
}
else if(root->right==nullptr){
//右子树空返回左子树,并删除结点
cur=root->left;
delete root;
return cur;
}
else{
//左右都不空,左子树去右子树最左边结点当儿子
//查找右子树最左结点
cur=root->right;
while(cur->left != nullptr){
cur=cur->left;
}
cur->left=root->left;
TreeNode* rig=root->right;
delete root;
return rig;
}
}
//情况3:根节点不空,小于要删除的结点,则查找右子树,递归直到情况2.
if(root->val > key) root->left = deleteNode(root->left,key);
//情况4:根节点不空,大于要删除的结点,则查找左子树,递归直到情况2.
if(root->val < key) root->right = deleteNode(root->right,key);
return root;
}
};