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;
};

猜你喜欢

转载自blog.csdn.net/romeo12334/article/details/84106462
今日推荐