删除二叉搜索树中的节点

在这里插入图片描述

class Solution {
public:
    TreeNode* deleteNode(TreeNode* root, int key) {
        //节点空
        if (root == nullptr) return root;
        //节点不空
        if (root->val == key) {
            //节点左子树和右子树都空
            if (root->left == nullptr && root->right == nullptr) {
                delete root;
                return nullptr;
            } else if (root->left == nullptr) {    //节点左子树空,右子树不空
                auto rroot = root->right;
                delete root;
                return rroot;
            } else if (root->right == nullptr) {   //节点右子树空,左子树不空
                auto lroot = root->left;
                delete root;
                return lroot;
            } else {                                       //节点左右节点都不空
                TreeNode* cur = root->right;
                while(cur->left != nullptr) {                 //找到右子树的最左节点
                    cur = cur->left;
                }
                cur->left = root->left;
                auto node = root;
                root = root->right;
                delete node;
                return root;
            }
        }
        if (root->val > key) root->left = deleteNode(root->left, key);
        if (root->val < key) root->right = deleteNode(root->right, key);
        return root; 
    }
};

时间复杂度:O(h)h为树的高度

思路:相较于BST的插入操作,删除更为复杂,有五种情况,但是大致框架都是一样,为有返回值的遍历边(用返回值构造树的某条边),所以时间复杂度为O(h)。
五种情况如下:
(1)没有删除的节点,则遍历到空节点直接返回。
(2)找到删除的节点,其左右儿子为空,则直接删除该节点,返回为空。
(3)找到删除的节点,其左空右不空,删除该节点,返回右节点(右节点代替了该节点)。
(4)找到删除的节点,其左不空右空,删除该节点,返回左节点(同理)。
(5)找到删除的节点,左右都不空,将左子树放到右子树的最左的节点上,删除该节点,返回右节点。

知识点:delete可要可不要,虽然不会报错但是加着最好,因为c++没有自动回收机制,而Java有。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值