给定二叉搜索树的根结点 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。
代码实现
C++:
题目的意思是要找出所有值在区间[L,R]内的节点,求它们的和。
所以核心问题就是查找。
LeetCode基础算法题第145篇:BST的范围总和
如上图,根据二叉搜索树,当L=6,R=15时,红框内就是我们要找的所有节点。
我们会发现,如果节点值小于等于6,没必要再查找它的左子树;同样的如果节点值大于等于15,没必要再查找它的右子树。
换句话说:
如果节点大于等于L,就需要查找左子树;
如果节点小于等于R,就需要查找右子树;
对于同时满足上述两个条件的节点,左右子树都需要查找。所以上述两个查找条件,可以作为并列的两个独立条件。
此外我们还需要另外的两个必要的判断:
当节点是空节点的时候,应该返回0,因为查找的时候会查找到叶子节点;
只有节点值在区间[L,R]内才统计它的值;
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int rangeSumBST(TreeNode* root, int L, int R) {
if(root == NULL){
return 0;
}
int sum = 0;
if(root->val >= L && root->val <= R){
sum = root->val;
}
//访问左树
if(root->val >= L){
sum += rangeSumBST(root->left, L, R);
}
//访问又树
if(root->val <= R){
sum += rangeSumBST(root->right, L, R);
}
return sum;
}
};
Python:
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def rangeSumBST(self, root, L, R):
"""
:type root: TreeNode
:type L: int
:type R: int
:rtype: int
"""
if not root:
return 0
sum = 0
if L <= root.val <= R:
sum = root.val
# 访问左子树
if root.val >= L:
sum += self.rangeSumBST(root.left, L, R)
# 访问右子树
if root.val <= R:
sum += self.rangeSumBST(root.right, L, R)
return sum