1.题目
给定二叉搜索树(BST)的根节点 root 和要插入树中的值 value ,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。
注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。
2.示例
root = [4,2,7,1,3], val = 5
输出:[4,2,7,1,3,5]
解释:另一个满足题目要求可以通过的树是:
3.答案
①递归
利用二叉树的搜索特性,这里选择一种简单插入方式,永远插入到叶子结点上。从上向下查找,找到元素应在位置,新建节点,更新节点的左右孩子。
TreeNode* insertIntoBST(TreeNode* root, int val) {
if(!root){ //如果节点为空,就新建val的节点插入
TreeNode*tmp=new TreeNode(val);
return tmp;
}
//val小于根,插入到左子树上,一定要更新左子树
if(val<root->val) root->left=insertIntoBST(root->left,val);
else if(val>root->val) root->right=insertIntoBST(root->right,val); //同理
return root;
}
②迭代
因为不存在返回寻找,只需从上到下找到插入元素硬插入的位置,这里使用par记录插入位置的父亲。当cur为空时就找到了val可以插入的位置,此时插入作为par的左孩子或者右孩子。
TreeNode* insertIntoBST(TreeNode* root, int val) {
if(!root){ //如果节点为空,就新建val的节点
TreeNode*tmp=new TreeNode(val);
return tmp;
}
TreeNode* par=root,*cur=root; //找到插入节点的父节点,从root开始
while(cur){
par=cur;
if(val<cur->val) cur=cur->left; //更新cur
else cur=cur->right;
}
if(val<par->val){ //par的左孩子
TreeNode* tmp=new TreeNode(val);
par->left=tmp;
}else{
TreeNode* tmp=new TreeNode(val);
par->right=tmp;
}
return root;
}
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/insert-into-a-binary-search-tree