-
Notifications
You must be signed in to change notification settings - Fork 523
/
Copy path15. Range Sum of BST.cpp
54 lines (46 loc) · 1.21 KB
/
15. Range Sum of BST.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
/*
Range Sum of BST
================
Given the root node of a binary search tree, return the sum of values of all nodes with a value in the range [low, high].
Example 1:
Input: root = [10,5,15,3,7,null,18], low = 7, high = 15
Output: 32
Example 2:
Input: root = [10,5,15,3,7,13,18,1,null,6], low = 6, high = 10
Output: 23
Constraints:
The number of nodes in the tree is in the range [1, 2 * 104].
1 <= Node.val <= 105
1 <= low <= high <= 105
All Node.val are unique.
*/
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution
{
void inorder(TreeNode *root, int low, int high, int &ans)
{
if (!root)
return;
inorder(root->left, low, high, ans);
if (root->val <= high && root->val >= low)
ans += root->val;
inorder(root->right, low, high, ans);
}
public:
int rangeSumBST(TreeNode *root, int low, int high)
{
int ans = 0;
inorder(root, low, high, ans);
return ans;
}
};