二叉搜索树的插入操作

1、描述

给定二叉搜索树（BST）的根节点和要插入树中的值，将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ，新值和原始二叉搜索树中的任意节点值都不同。

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

来源：力扣（LeetCode）
链接：
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2、关键字

二叉搜索树，插入

3、思路

递归

4、notes

递归出口，返回值类型

5、复杂度

时间：O（N),最坏需要遍历到最后
空间：O（1）

6、code

/**
 * 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* insertIntoBST(TreeNode* root, int val) {
        if(!root) return new TreeNode(val); // 如果为空，直接构造，
        if(root->val < val){ //去右边，递归调用右边，，使用右孩子，
            root->right = insertIntoBST(root->right, val);
        }
        else{  // 递归调用左边，使用左孩子，
            root->left = insertIntoBST(root->left, val);
        }
        return root;                
    }
};


// 迭代解法
class Solution {
public:
    TreeNode* insertIntoBST(TreeNode* root, int val) {
        if (root == nullptr) {
            return new TreeNode(val);
        }
        TreeNode* pos = root;
        while (pos != nullptr) {
            if (val < pos->val) {
                if (pos->left == nullptr) {
                    pos->left = new TreeNode(val);
                    break;
                } else {
                    pos = pos->left;
                }
            } else {
                if (pos->right == nullptr) {
                    pos->right = new TreeNode(val);
                    break;
                } else {
                    pos = pos->right;
                }
            }
        }
        return root;
    }
};

7 、变形：如果接口是node

为了实现“伪”map，红黑树----> 使用二叉搜索树实现。

/**
 * 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:
    void insert(TreeNode * root,TreeNode* node){
        if(!root) root = node;
  		// 
        if(root->left == nullptr && root->val > node->val){
            root->left = node; 
            return;
        }
        else if(root->right == nullptr && root->val < node->val){
            root->right = node;
            return;
        }
        else if(root->left && root->val > node->val){
            insert(root->left,node); 
        }
        else if(root->right && root->val < node->val){
            insert(root->right,node);
        }
  
    }

    TreeNode* insertIntoBST(TreeNode* root, int val) {
        if(!root) return new TreeNode(val);
        TreeNode * node = new TreeNode(val);
        insert(root,node);
        return root;
                     
    }
};