LeetCode——938 二叉搜尋樹的範圍和
阿新 • • 發佈:2018-11-23
給定二叉搜尋樹的根結點 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;
};