你需要保证该树仍然是一棵二叉查找树。
样例:给出如下一棵二叉查找树,在插入节点6之后这棵二叉查找树可以是这样的:
2 2
/ \ / \
1 4 --> 1 4
/ / \
3 3 6
解题思路:要在二叉查找树的定义,如果要添加的节点值比根节点小,则把它添加到根节点的左子树,若要添加的节点值比根节点值要大,则把它添加到根节点的右子树,具体实现方法可以用递归的方法实现;
实现代码:
/**
* 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;
else if(node->val<root->val)
root->left=insertNode(root->left,node);
else
root->right=insertNode(root->right,node);
}
return root;
}
};
做题感谢:感觉这个题最重要的返回值类型,因为它返回的是节点,不是节点值,也不是空类型。最开始写的时候,我在设想是不是最后可以借助中序遍历把一整棵树直接输出来,后来一想这跟递归函数不符,最终采用了这种方式。