算法 -【二叉搜索树的范围和】

题目

给定二叉搜索树的根结点 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 x 10⁴] 内
• 1 <= Node.val <= 10⁵
• 1 <= low <= high <= 10⁵
• 所有 Node.val 互不相同

分析

这题让计算二叉搜索树中节点值位于[low,high]之间的所有节点之和。一种常规的解决思路就是遍历这棵树的所有节点，然后累加符合条件的节点值。

但这题说的是二叉搜索树，在二叉搜索树中左子树的值都小于根节点，右子树的值都大于根节点，并且每一棵子树也都满足这个条件，我们可以根据这个条件对这题做个优化。

因为右子树的值都比根节点大，所以如果根节点大于high，说明根节点的右子树也都大于high，不满足条件。只有在根节点值小于high的时候，右子树的有些值才有可能小于high。左子树同理。

代码

public int rangeSumBST(TreeNode root, int low, int high) {
    if (root == null)
        return 0;
    int res = 0;
    if (root.val >= low && root.val <= high)
        res += root.val;

    if (root.val < high)
        res += rangeSumBST(root.right, low, high);
    if (root.val > low)
        res += rangeSumBST(root.left, low, high);

    return res;
}

原文：https://mp.weixin.qq.com/s/G_53I00-pycgFQNLAdTBaQ