在二叉查找树中插入节点
问题描述:
给定一棵二叉查找树和一个新的树节点,将节点插入到树中。
你需要保证该树仍然是一棵二叉查找树。
样例给出如下一棵二叉查找树,在插入节点6之后这棵二叉查找树可以是这样的:
2 2
/ \ / \
1 4 --> 1 4
/ / \
3 3 6
解题思路:
由题目可知,插入的结点一定为叶子结点,所以利用递归的思想,若要插入的值小于根结点,则插到根结点的左子树中,否则,插入到根结点的右子树中。
代码实现:
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) root=node;
else {
if(root->val>node->val) root->left=insertNode(root->left,node);
else root->right=insertNode(root->right,node);
}
return root;
}
};
A题感悟:
此题较简单,只需要把插入的结点和原树连接起来即可。