二叉搜索树中的插入操作
给定二叉搜索树(BST)的根节点和要插入树中的值,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据保证,新值和原始二叉搜索树中的任意节点值都不同。
注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回任意有效的结果。
1、给定二叉搜索树: 4 / \ 2 7 / \ 1 3 和 插入的值: 5
返回这个二叉搜索树:
4 / \ 2 7 / \ / 1 3 5
利用二叉搜索树的定义:
- 每个节点中的值必须
大于
(或等于)存储在其左侧子树中的任何值。 - 每个节点中的值必须
小于
(或等于)存储在其右子树中的任何值。
递归实现
public TreeNode insertIntoBST(TreeNode root, int val) {
if(root == null)
return new TreeNode(val);
if(root.val > val)
root.left = insertIntoBST(root.left, val);
else
root.right = insertIntoBST(root.right, val);
return root;
}
从上到下跑一遍,与每个节点比较,
public TreeNode insertIntoBST(TreeNode root, int val) {
if(root == null)
return new TreeNode(val);
TreeNode tnode = root;
while(tnode != null){
if(tnode.val > val){
if(tnode.left == null){
tnode.left = new TreeNode(val);
break;
}
else
tnode = tnode.left;
}
else{
if(tnode.right == null){
tnode.right = new TreeNode(val);
break;
}
else
tnode = tnode.right;
}
}
return root;
}