【算法刷怪】leetcode-450.删除二叉搜索树中的结点

非最优解,仅学习记录,参考了题解的思路。

算法思路:

情况大概几种:

  1. 根节点为空,则直接返回根节点
  2. 根节点不空,正好是要删除的结点,判断左右子树是否为空
    • 都为空则直接删除
    • 一边不空则子树直接变根结点
    • 两边都不空则左子树去右子树最左结点下
  3. 根节点不空,小于要删除的结点,则查找右子树,递归直到情况2
  4. 根节点不空,大于要删除的结点,则查找左子树,递归直到情况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;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值