701. 二叉搜索树中的插入操作

最新推荐文章于 2024-03-12 21:30:47 发布

碰碰狗

最新推荐文章于 2024-03-12 21:30:47 发布

阅读量55

点赞数

分类专栏： Leetcode刷题笔记（C++）文章标签： leetcode 二叉搜索树递归

本文链接：https://blog.csdn.net/suitaox26456/article/details/127436612

版权

Leetcode刷题笔记（C++）专栏收录该内容

29 篇文章 0 订阅

订阅专栏

方法一：迭代

class Solution
{
public:
    TreeNode* insertIntoBST(TreeNode* root, int val)
    {
        TreeNode* parent = root;
        TreeNode* child = root;

        if (root == nullptr)
        {
            root = new TreeNode{ val };
            return root;
        }

        while (child != nullptr)
        {
            parent = child;

            if (val < child->val)
            {
                child = child->left;
            }
            else if (val > child->val)
            {
                child = child->right;
            }
            else
            {
                return root;
            }
        }

        child = new TreeNode{ val };

        if (val > parent->val)
        {
            parent->right = child;
        }
        else
        {
            parent->left = child;
        }

        return root;
    }
};

方法二：递归

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

        return root;
    }
};