力扣常规题：删除二叉搜索树中的节点

c栈算法小辰哥

于 2024-07-13 19:32:26 发布

阅读量267

点赞数 4

分类专栏：漫话算法文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/whcwhc111111/article/details/140404946

版权

漫话算法专栏收录该内容

129 篇文章 1 订阅

订阅专栏

1.450. 删除二叉搜索树中的节点 - 力扣（LeetCode）

常规操作

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
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)
    {
        return nullptr;
    }
    else if(root->left==nullptr)
    {
        return root->right;
    }
    else if(root->right==nullptr)
    {
        return root->left;
    }
    else{
        TreeNode*cur=root->right;
        while(cur->left!=nullptr)
        {
            cur=cur->left;
        }
        cur->left=root->left;
        TreeNode*tmp=root;
        root=root->right;
        delete tmp;
        return root;
    }
}
if(root->val>key)
{
    root->left=deleteNode(root->left,key);
}
if(root->val<key)
{
    root->right=deleteNode(root->right,key);
}
return root;
    }
};