一、题目内容 中等
给定二叉搜索树(BST)的根节点 root
和要插入树中的值 value
,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。
注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果
示例1:
输入:root = [4,2,7,1,3], val = 5 输出:[4,2,7,1,3,5] 解释:另一个满足题目要求可以通过的树是:
示例2:
输入:root =
[40,20,60,10,30,50,70]
, val = 25 输出:[40,20,60,10,30,50,70,null,null,25]
示例3:
输入:root =
[4,2,7,1,3,null,null,null,null,null,null]
, val = 5 输出:[4,2,7,1,3,5]
二、解题思路
我们按照这个图的解题思路来想。
- 5 > 4,向右子树走到 7
- 5 < 7,向左子树走,是空的,5 就作为 7 的左子节点
三、具体代码
/**
* @param {TreeNode} root
* @param {number} val
* @return {TreeNode}
*/
var insertIntoBST = function (root, val) {
if (!root) return new TreeNode(val)
if (root.val < val) root.right = insertIntoBST(root.right, val)
if (root.val > val) root.left = insertIntoBST(root.left, val)
return root
};
四、其他解法
/**
* @param {TreeNode} root
* @param {number} val
* @return {TreeNode}
*/
var insertIntoBST = function (root, val) {
if (!root) return new TreeNode(val)
const queue = [root]
while (queue.length) {
const node = queue.shift()
if (node.val < val) {
if (!node.right) {
node.right = new TreeNode(val)
break
}
queue.push(node.right)
} else {
if (!node.left) {
node.left = new TreeNode(val)
break
}
queue.push(node.left)
}
}
return root
};