删除二叉搜索树中的节点

给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。

一般来说,删除节点可分为两个步骤:

首先找到需要删除的节点;
如果找到了,删除它。
说明: 要求算法时间复杂度为 O(h),h 为树的高度。

思路:先根据二叉搜索树的性质,找到要删除的节点。若该节点没有右子树,直接用该节点的左子树代替该节点;若该节点没有左子树,直接用该节点的右子树代替该节点。若节点既有左子树又有右子树,把节点的左子树作为右子树的最左子树的左子树;也可以把节点的右子树作为左子树的最右子树的右子树。看上去有点绕,多理解一下

TreeNode* deleteNode(TreeNode* root, int key) {
    if(!root)return nullptr;
    if(key<root->val)root->left=deleteNode(root->left,key);
    else if(key>root->val)root->right=deleteNode(root->right,key);
    else{
        if(!root->right)
            return root->left;
        if(!root->left)
            return root->right;

        TreeNode* r=root->right;
        //找到右子树的最左子树
        while(r->left)r=r->left;
        r->left=root->left;
        root=root->right;
    }
    return root;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
C语言中的二搜索树是一种非常常用的数据结构,其插入和查找的时间复杂度都是O(logn)。当然,在使用二搜索树时,我们也需要注意它的删除操作,否则会影响它的性能。 二搜索树的删除操作可以分为以下三种情况: 1. 被删除节点为叶子节点:直接将其删除即可。 2. 被删除节点只有一个子节点:将其子节点替代被删除节点即可。 3. 被删除节点有两个子节点:需要找到其右子树中的最小值或左子树中的最大值来替代被删除节点。 具体实现时,我们可以使用递归或者迭代的方式进行实现。下面是一个使用递归的实现方式: ``` struct TreeNode* deleteNode(struct TreeNode* root, int key) { if (root == NULL) return root; if (key < root->val) { root->left = deleteNode(root->left, key); } else if (key > root->val) { root->right = deleteNode(root->right, key); } else { if (root->left == NULL) { struct TreeNode* temp = root->right; free(root); return temp; } else if (root->right == NULL) { struct TreeNode* temp = root->left; free(root); return temp; } else { struct TreeNode* temp = findMin(root->right); root->val = temp->val; root->right = deleteNode(root->right, temp->val); } } return root; } struct TreeNode* findMin(struct TreeNode* node) { while (node->left != NULL) node = node->left; return node; } ``` 以上代码中,deleteNode函数使用递归实现了删除操作,findMin函数则是用来找到右子树中的最小值。在deleteNode函数中,我们首先判断了要删除节点是否为空,然后再根据二搜索树的性质来判断要删除节点在左子树还是右子树中。如果要删除节点为叶子节点或只有一个子节点,则直接将其替换即可;如果要删除节点有两个子节点,则需要找到右子树中的最小值(或左子树中的最大值),将其赋值给被删除节点,并在右子树中删除该最小值。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值