【每日刷题】二叉搜索树中的插入操作

题目地址

https://leetcode-cn.com/problems/insert-into-a-binary-search-tree/

题目描述：二叉搜索树中的插入操作

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

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

例：

给定二叉搜索树:

       4
      / \
     2   7
    / \
   1   3

以及待插入的值: 5

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

        4
      /   \
     2     7
    / \   /
   1   3 5

或者这个树也是有效的:

        5
      /   \
     2     7
    / \   
   1   3
        \
         4

解答

从题目描述中可以看出，存在多种不同的插入方式。

这里采用的方法是：依据 BST 的定义，首先找到待插入结点适合的位置，然后将结点插入其中。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    
    TreeNode* FindPostion( TreeNode* root, int val){
        TreeNode* pre = root;
        while( root){
            pre = root;
            if(root->val > val)
                root = root->left;
            else
                root = root->right;
        }
        return pre;
    }
    TreeNode* insertIntoBST(TreeNode* root, int val) {
        TreeNode* pre = FindPostion( root, val);
        
        TreeNode* insertNode = new TreeNode( val);
        
        if( pre->val > val)
            insertNode->left = pre->left,   pre->left = insertNode;
        else
            insertNode->right = pre->right, pre->right = insertNode;
        
        return root;
    }
};