在二叉查找树中插入节点

问题描述：

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

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

 注意事项

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

样例

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

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

解题思路：

二叉查找树的特点是左子树上节点的数值都小于根节点的数值，右子树上节点的数值都大于等于根节点的数值。插入节点的关键是找到插入节点的位置，即判断插入节点的数值与树的节点上数值的大小的关系，当找到插入节点的位置后直接将插入的节点放入树中返回root即可。从根节点开始建立循环，比较插入节点的数值与树上节点值得大小。插入节点值为t，如果t小于根节点的数值，则插入节点的位置在左子树，否则在右子树，依次向下循环直到叶子节点。新建节点pre用以保存前面的节点，退出循环后的pre节点就是插入的位置。

代码实现：

/**
 * 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.
     */
    TreeNode* insertNode(TreeNode* root, TreeNode* node) {
        // write your code here
        if(root==NULL) return node;
        if(node==NULL) return root;
        int t=node->val;
        TreeNode* p=root;
        TreeNode* pre=p;
        while(p!=NULL){
            pre=p;
            if(t>=p->val){
              p=p->right;
            }
            else 
              p=p->left;
            
        }
        if(pre->val>t)
            pre->left=node;
        if(pre->val<=t)
            pre->right=node;
        return root;
    }
};

解题感悟：

找到插入节点的位置是难点，一开始没有想到要保存循环体中的前一个节点。