[LeetCode] 938. 二叉搜索树的范围和

1 题目描述

给定二叉搜索树的根结点 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。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/range-sum-of-bst
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2 解题思路

我们对树进行深度优先搜索，对于当前节点 node，

• 如果 node.val 恰好在题目要求的区间里面，我们就要把这个值加入到我们设置的临时变量中
• 如果 node.val 小于等于 L，那么只需要继续搜索它的右子树；
• 如果 node.val 大于等于 R，那么只需要继续搜索它的左子树；如果 node.val 在区间 (L, R)中，则需要搜索它的所有子树。

下面的代码是采用的深度优先搜索的递归方法

3 解决代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    int ans;
    public int rangeSumBST(TreeNode root, int L, int R) {
        ans = 0;
        dfs(root, L,R);
        return ans;  
    }
    public void dfs(TreeNode node,int L,int R){
        if(node != null){
            //如果是刚刚好在这个区间内，直接加上就可以
            if(L<=node.val && node.val<=R){
                ans = ans + node.val;
            }
            //如果左边的值小于区间，那么接着往下找
            if(L<node.val){
                dfs(node.left, L, R);
            }
            //同理
            if(node.val < R){
                dfs(node.right, L, R);
            }
        }
    }
}