给定二叉搜索树的根结点 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
所限定的范围进行"去枝".
题解:
/**
* 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 sum = 0;
void order(TreeNode* root, int L, int R) {
if (root!=NULL) {
if (root->val>=L&&root->val<=R) { //判断范围
sum += root->val;
}
if (root->val<=L) { //去枝
order(root->right, L, R);
} else if (root->val>=R) {
order(root->left, L, R);
} else {
order(root->left, L, R);
order(root->right, L, R);
}
}
}
int rangeSumBST(TreeNode* root, int L, int R) {
order(root, L, R);
return sum;
}
};
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/range-sum-of-bst