给定二叉搜索树的根结点 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
。
题目涉及到二叉搜索树而不是普通的二叉树,那么在考虑问题的时候,就一定要联想到二叉搜索树的性质。根据题目描述,首先想到的就是遍历整棵树,找到给定区间里的值求和即可,这是普通二叉树的做法,既然是二叉搜索树,那么当我们找到一个节点比L小的时候,那么以这个节点为根的子树就不需要遍历了,因为二叉搜索树的左孩子(子树)小于根,右侧的R也是同样的道理,下面是AC代码:
Java:
public int rangeSumBST(TreeNode root, int L, int R) {
if (root == null) {
return 0;
}
if (root.val < L) {
return rangeSumBST(root.right, L, R);
}
if (root.val > R) {
return rangeSumBST(root.left, L, R);
}
return root.val + rangeSumBST(root.right, L, R) + rangeSumBST(root.left, L, R);
}
Python:
def rangeSumBST(self, root, L, R):
"""
:type root: TreeNode
:type L: int
:type R: int
:rtype: int
"""
if root is None: return 0
if root.val < L:
return self.rangeSumBST(root.right, L, R)
if root.val > R:
return self.rangeSumBST(root.left, L, R)
return root.val + self.rangeSumBST(root.left, L, R) + self.rangeSumBST(root.right, L, R)