LeetCode——938 二叉搜索树的范围和

给定二叉搜索树的根结点 root,返回 L 和 R(含)之间的所有结点的值的和。

二叉搜索树保证具有唯一的值。

示例 1:

输入:root = [10,5,15,3,7,null,18], L = 7, R = 15
输出:32

示例 2:

输入:root = [10,5,15,3,7,13,18,1,null,6], L = 6, R = 10
输出:23

提示:
树中的结点数量最多为 10000 个。
最终的答案保证小于 2^31。


思路

二叉搜索树使用中序遍历就可以得到递增数组,所以先用中序遍历得到一个递增数组。

  let inorderlist = []
  const inorder = function (root) {
    if (!root) return;
    inorder(root.left);
    inorderlist.push(root.val);
    inorder(root.right);
  }

然后循环遍历数组将 L 和 R 之间的数加起来。

  let sum = 0;
  for (let val of inorderlist) {
    if (val < L) continue;
    else if (val >= L && val <= R) {
      sum += val;
    } else {
      break;
    }
  }

完整代码

/** * Definition for a binary tree node. * function TreeNode(val) { * this.val = val; * this.left = this.right = null; * } */
/** * @param {TreeNode} root * @param {number} L * @param {number} R * @return {number} */
var rangeSumBST = function(root, L, R) {
  let inorderlist = []
  const inorder = function (root) {
    if (!root) return;
    inorder(root.left);
    inorderlist.push(root.val);
    inorder(root.right);
  }
  inorder(root);
  let sum = 0;
  for (let val of inorderlist) {
    if (val < L) continue;
    else if (val >= L && val <= R) {
      sum += val;
    } else {
      break;
    }
  }
  return sum;
};
全部评论

相关推荐

10-05 23:02
东北大学 Java
我说句实话啊:那时候看三个月培训班视频,随便做个项目背点八股,都能说3 40w是侮辱价
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务