题目描述
给定二叉搜索树的根结点 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 个。
最终的答案保证小于 。
思路:
一看到二叉树,很自然会想到递归。
- 我们需要搞清除两点:1.递归结束的条件 2.如果不满足递归结束条件做何操作
- 递归结束条件
if(root == null) return 0;
- 子问题
例如我们要求这棵树的范围和,可以假设节点10的左右子树的范围和都算出来了(通过递归),此时如果当前节点10范围刚好在L
和R
之间,就直接加上去。
return root.val + rangeSumBST(root.right, L, R) + rangeSumBST(root.left, L, R);
如果root.val < L
,那么root
的左子树都不可能在L
和R
之间,遍历右子树即可;反之,root.val > R
,遍历左子树。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public int rangeSumBST(TreeNode root, int L, int R) {
if(root == null) return 0;
if(root.val < L){
return rangeSumBST(root.right, L, R);
}else if(root.val > R){
return rangeSumBST(root.left, L, R);
}
return root.val + rangeSumBST(root.right, L, R) + rangeSumBST(root.left, L, R);
}
}
或者从底到顶递归遍历的方式累加
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public int rangeSumBST(TreeNode root, int L, int R) {
return rangeSumBST(root, L, R, new int[1]);
}
private int rangeSumBST(TreeNode root, int L, int R, int[] sum){
if(root == null) return 0;
rangeSumBST(root.left, L, R, sum);
rangeSumBST(root.right, L, R, sum);
if(root.val >= L && root.val <= R){
sum[0] += root.val;
}
return sum[0];
}
}