刷题记录第七十九天-删除二叉搜索树中的节点

onlyzzr

于 2024-03-19 11:54:22 发布

阅读量112

点赞数 2

分类专栏：每日一题文章标签： java 算法数据结构

本文链接：https://blog.csdn.net/weixin_45850570/article/details/136837920

版权

每日一题专栏收录该内容

109 篇文章 0 订阅

订阅专栏

该代码片段展示了如何在C++中使用递归方法实现二叉搜索树的删除节点功能。当给定键值时，函数会找到并移除相应节点，确保树的结构保持有序。

摘要由CSDN通过智能技术生成

class Solution {
public:
    TreeNode* deleteNode(TreeNode* root, int key) {
        if(root==nullptr) return nullptr;
        if(root->val>key){
            root->left = deleteNode(root->left,key);
        }else if(root->val<key){
            root->right = deleteNode(root->right,key);
        }else{
            if(!root->left&&!root->right){
                return nullptr;
            }
            if(root->left==nullptr){
                return root->right;
            }
            TreeNode* right = root->right;
            TreeNode* node = root->left;
            while(node->right){
                node=node->right;
            }
            node->right = right;
            root = root->left;
        }
        return root;
    }
};