一、题目描述
给定二叉搜索树的根结点 root,返回值位于范围 [low, high] 之间的所有结点的值的和。
示例 1:
输入:root = [10,5,15,3,7,null,18], low = 7, high = 15
输出:32
示例 2:
输入:root = [10,5,15,3,7,13,18,1,null,6], low = 6, high = 10
输出:23
提示:
树中节点数目在范围 [1, 2 * 104] 内
1 <= Node.val <= 105
1 <= low <= high <= 105
所有 Node.val 互不相同
二、题目解法
1、中序遍历
二叉搜索树中序遍历可以得到一个递增的集合,根据这个特点,就可以计算出符合规则的节点值和
public int rangeSumBST(TreeNode root, int low, int high) {
if (root == null)
return 0;
int current = root.val <= high && root.val >= low ? root.val : 0;
return rangeSumBST(root.left, low, high) + current + rangeSumBST(root.right, low, high);
}
2、中序遍历 + 剪枝
上个版本中,递归遍历了二叉搜索树中的所有节点,所以,有没有方式可以优化呢,只遍历符合条件的节点值?
只要在递归过程中,判断节点值是否满足要查询的节点值的范围之内即可,这样,就可以去掉不符合题意的节点值的递归
java版:
public int rangeSumBST(TreeNode root, int low, int high) {
if (root == null)
return 0;
var sum = 0;
if (root.val > low)
sum += rangeSumBST(root.left, low, high);
sum += root.val <= high && root.val >= low ? root.val : 0;
if (root.val < high)
sum += rangeSumBST(root.right, low, high);
return sum;
}
javascript版:
var rangeSumBST = function(root, low, high) {
if (!root)
return 0
let sum = 0
if (root.val >= low)
sum += rangeSumBST(root.left, low, high)
sum += root.val <= high && root.val >= low ? root.val : 0
if (root.val <= high)
sum += rangeSumBST(root.right, low, high)
return sum
};