题目描述:
给定二叉搜索树(BST)的根节点和要插入树中的值,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据保证,新值和原始二叉搜索树中的任意节点值都不同。
注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回任意有效的结果。
例如,
给定二叉搜索树:
4
/ \
2 7
/ \
1 3
和 插入的值: 5
你可以返回这个二叉搜索树:
4
/ \
2 7
/ \ /
1 3 5
递归:
class Solution {
public:
TreeNode* insertIntoBST(TreeNode* root, int val) {
if(root==NULL)
return new TreeNode(val);
if(root->val<val){
root->right=insertIntoBST(root->right,val);
}
else root->left=insertIntoBST(root->left,val);
return root;
}
};
迭代:
class Solution {
public:
TreeNode* insertIntoBST(TreeNode* root, int val) {
if(root==NULL)
return new TreeNode(val);
TreeNode* parent=root;
TreeNode* p=root;
while(p!=NULL){
parent=p;
p=p->val<val?p->right:p->left;
}
if(parent->val<val){
parent->right=new TreeNode(val);
}
else parent->left=new TreeNode(val);
return root;
}
};