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

二叉搜索树中的插入操作

题目描述

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

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

例如,

给定二叉搜索树:

    4
   / \
  2   7
 / \
1   3

和 插入的值: 5

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

     4
   /   \
  2     7
 / \   /
1   3 5

或者这个树也是有效的:

     5
   /   \
  2     7
 / \   
1   3
     \
      4

解题思路

递归

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;
    }
};