938. 二叉搜索树的范围和

题目描述

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

思路

• 简单递归~
  判断当前值和目标区间的大小关系。
• 如果当前节点 val < low，那么递归进入右子树。 如果 val > high，那么递归进入左子树。 如果位于 low 和 high 之间，那么返回 当前 val + 左子树递归值 + 右子树递归值。
• 时间复杂度为O(n)

代码

class Solution {
    public int rangeSumBST(TreeNode root, int low, int high) {
    	if (root == null)
    		return 0;
    	//左子树
    	if(root.val > high) {
    		return rangeSumBST(root.left, low, high);
    	}
    	//右子树
    	else if(root.val < low) {
    		return rangeSumBST(root.right, low, high);
    	}
    	//当前节点 + 左子树 + 右子树
    	else {
    		return root.val + rangeSumBST(root.left, low, root.val-1) + rangeSumBST(root.right, root.val + 1, high);
    	}
    }
}