题目描述:
给定二叉搜索树的根结点 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/range-sum-of-bst
解题思路:
运用递归的思想,因为题目给的是一棵二叉搜索树,所以可以直接判断如果比low值小的就在左子树找,比high大的,在右子树找,直到结点为空。
代码:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public int rangeSumBST(TreeNode root, int low, int high) {
if(root==null){
return 0;
}
//如果节点的值在范围之间,就加起来,递归寻找符合的值。
if(root.val>=low && root.val<=high){
return root.val+rangeSumBST(root.right,low,high)+rangeSumBST(root.left,low,high);
}
//比low小,在右边找
else if(root.val<low){
return rangeSumBST(root.right,low,high);
}
//比high大,在左边找
else if(root.val>high){
return rangeSumBST(root.left,low,high);
}
return 0;
}
}