一、题目内容 中等

给定二叉搜索树(BST)的根节点 root 和要插入树中的值 value ,将值插入二叉搜索树。 返回插入后二叉搜索树的根节点。 输入数据 保证 ,新值和原始二叉搜索树中的任意节点值都不同。

注意,可能存在多种有效的插入方式,只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果

示例1:

20. 二叉搜索树中的插入操作(701) - 图1 输入:root = [4,2,7,1,3], val = 5 输出:[4,2,7,1,3,5] 解释:另一个满足题目要求可以通过的树是: 20. 二叉搜索树中的插入操作(701) - 图2

示例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]

二、解题思路

20. 二叉搜索树中的插入操作(701) - 图3
我们按照这个图的解题思路来想。

  1. 5 > 4,向右子树走到 7
  2. 5 < 7,向左子树走,是空的,5 就作为 7 的左子节点

    三、具体代码

    1. /**
    2. * @param {TreeNode} root
    3. * @param {number} val
    4. * @return {TreeNode}
    5. */
    6. var insertIntoBST = function (root, val) {
    7. if (!root) return new TreeNode(val)
    8. if (root.val < val) root.right = insertIntoBST(root.right, val)
    9. if (root.val > val) root.left = insertIntoBST(root.left, val)
    10. return root
    11. };

    四、其他解法

    1. /**
    2. * @param {TreeNode} root
    3. * @param {number} val
    4. * @return {TreeNode}
    5. */
    6. var insertIntoBST = function (root, val) {
    7. if (!root) return new TreeNode(val)
    8. const queue = [root]
    9. while (queue.length) {
    10. const node = queue.shift()
    11. if (node.val < val) {
    12. if (!node.right) {
    13. node.right = new TreeNode(val)
    14. break
    15. }
    16. queue.push(node.right)
    17. } else {
    18. if (!node.left) {
    19. node.left = new TreeNode(val)
    20. break
    21. }
    22. queue.push(node.left)
    23. }
    24. }
    25. return root
    26. };