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
解题思路:首先插入的结点一定是叶子节点,所以我要找到要正确插入的位置,主要是有三种情况,走到叶子节点,比大小,该左还是右,走到有一个孩子节点为空的结点,在判断是该直接插入,还是在进入相应的不空的结点子树。
class Solution {
public:
TreeNode* insertIntoBST(TreeNode* root, int val) {
IntoBst(root,val);
return root;
}
void IntoBst(TreeNode* p, int val){
if(!p->left&&!p->right){
if(p->val<val){
p->right=new TreeNode(val);
}
else{
p->left=new TreeNode(val);
}
return;
}
if(p->val>val){
if(!p->left){
p->left=new TreeNode(val);
return;
}
else IntoBst(p->left,val);
}
if(p->val<val){
if(!p->right){
p->right=new TreeNode(val);
return;
}
else IntoBst(p->right,val);
}
}
};