给定二叉搜索树(BST)的根节点和要插入树中的值,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据保证,新值和原始二叉搜索树中的任意节点值都不同。
注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回任意有效的结果。
例如,
给定二叉搜索树:
4
/ \
2 7
/ \
1 3
和 插入的值: 5
你可以返回这个二叉搜索树:
4
/ \
2 7
/ \ /
1 3 5
或者这个树也是有效的:
5
/ \
2 7
/ \
1 3
\
4
分析:方法一:递归(通过递归 更改左右子树,BST树的特质,顺序的,左边小,右边大)
代码:
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;
}
方法二:迭代,每次判断一下,当前节点 和 操作节点的关系,再往下做
public TreeNode insertIntoBST(TreeNode root, int val) {
//边界条件判断
if (root == null)
return new TreeNode(val);
TreeNode cur = root;
while (true) {
//如果当前节点cur的值大于val,说明val值应该插入到
//当前节点cur的左子树,否则就插入到当前节点cur的右子树
if (cur.val > val) {
//如果左子节点不为空,就继续往下找
if (cur.left != null) {
cur = cur.left;
} else {//如果左子节点为空,就直接插入去,然后再返回root节点
cur.left = new TreeNode(val);
return root;
}
} else {//同上
if (cur.right != null) {
cur = cur.right;
} else {
cur.right = new TreeNode(val);
return root;
}
}
}
}