LeetCode701. 二叉搜索树中的插入操作(C++)

最新推荐文章于 2024-05-04 19:26:32 发布

不重复啊

最新推荐文章于 2024-05-04 19:26:32 发布

阅读量256

点赞数

分类专栏： C++ LeetCode 文章标签：二叉树 leetcode

本文链接：https://blog.youkuaiyun.com/visual568/article/details/107831237

版权

LeetCode 同时被 2 个专栏收录

12 篇文章

订阅专栏

C++

6 篇文章

订阅专栏

本文详细介绍了如何在二叉搜索树(BST)中插入新值，并提供了递归和广度优先搜索两种方法的实现代码。通过实例演示了插入操作后的树结构变化，帮助读者理解二叉搜索树的动态调整过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

二叉搜索树中的插入操作

题目描述

给定二叉搜索树（BST）的根节点和要插入树中的值，将值插入二叉搜索树。返回插入后二叉搜索树的根节点。保证原始二叉搜索树中不存在新值。

注意，可能存在多种有效的插入方式，只要树在插入后仍保持为二叉搜索树即可。你可以返回任意有效的结果。

例如,

给定二叉搜索树:

和插入的值: 5

你可以返回这个二叉搜索树:

或者这个树也是有效的:

解题思路

递归

class Solution {
public:
    void dfs(TreeNode* root,int val)
    {
        if(root==nullptr)
            return;
        if(root->val>val)
        {
            dfs(root->left,val);
            if(!root->left)
                root->left=new TreeNode(val);
        }
        else if(root->val<val)
        {
            dfs(root->right,val);
            if(!root->right)
                root->right=new TreeNode(val);
        }
    }
    TreeNode* insertIntoBST(TreeNode* root, int val) {
        if(root)
        dfs(root,val);
        else
        root=new TreeNode(val);
        return root;
    }
};

广度优先搜索

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