在二叉查找树中插入节点-LintCode

描述：

给定一棵二叉查找树和一个新的树节点，将节点插入到树中。

你需要保证该树仍然是一棵二叉查找树。

 注意事项

You can assume there is no duplicate values in this tree + node.

样例：

给出如下一棵二叉查找树，在插入节点6之后这棵二叉查找树可以是这样的：

  2             2
 / \           / \
1   4   -->   1   4
   /             / \ 
  3             3   6

思路：

若二叉排序树为空树，则新插入的结点为新的根节点；否则，新插入的结点必为一个新的叶子结点。

这样，我们就只需要比较给定node与根结点的值，从而选择继续访问根结点的左子树还是右子树。

访问到空位置，即插入node。

AC代码：

/**
 * Definition of TreeNode:
 * class TreeNode {
 * public:
 *     int val;
 *     TreeNode *left, *right;
 *     TreeNode(int val) {
 *         this->val = val;
 *         this->left = this->right = NULL;
 *     }
 * }
 */
class Solution {
public:
    /**
     * @param root: The root of the binary search tree.
     * @param node: insert this node into the binary search tree
     * @return: The root of the new binary search tree.
     */
    void insertNodeCore(TreeNode* root,TreeNode* node)
    {
        if(root->val > node->val && root->left==NULL)
           {
               root->left=node;
               return;
           }
        if(root->val < node->val && root->right==NULL)
           {
               root->right=node;
               return;
           }
        if(root->val > node->val)
            insertNodeCore(root->left,node);
        else
            insertNodeCore(root->right,node);
        
    }
    
    TreeNode* insertNode(TreeNode* root, TreeNode* node) {
        // write your code here
        if(root==NULL)
        return node;
        insertNodeCore(root,node);
        return root;
    
    }
};