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