给定二叉搜索树(BST)的根节点和要插入树中的值,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。
输入数据保证,新值和原始二叉搜索树中的任意节点值都不同。注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回任意有效的结果。
例如,
给定二叉搜索树:
4 / \ 2 7 / \ 1 3
和 插入的值: 5
你可以返回这个二叉搜索树:4 / \ 2 7 / \ / 1 3 5 或者这个树也是有效的: 5 / \ 2 7 / \ 1 3 \ 4
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/insert-into-a-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
不要求红黑树的话毫无难度(如果要求红黑树无非是抄代码呗四舍五入也毫无难度)
递归解法:
class Solution {
public TreeNode insertIntoBST(TreeNode root, int val) {
if(root == null) {
return new TreeNode(val, null, null);
}
if(val > root.val) {
if(root.right == null) {
root.right = new TreeNode(val, null, null);
} else {
insertIntoBST(root.right, val);
}
} else {
if(root.left == null) {
root.left = new TreeNode(val, null, null);
} else {
insertIntoBST(root.left, val);
}
}
return root;
}
}
迭代解法:
class Solution {
public TreeNode insertIntoBST(TreeNode root, int val) {
if (root == null) {
return new TreeNode(val);
}
TreeNode pos = root;
while (true) {
if (val < pos.val) {
if (pos.left == null) {
pos.left = new TreeNode(val);
break;
} else {
pos = pos.left;
}
} else {
if (pos.right == null) {
pos.right = new TreeNode(val);
break;
} else {
pos = pos.right;
}
}
}
return root;
}
}