-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathMinimumBSTSumsubtree.cpp
More file actions
39 lines (36 loc) · 1.21 KB
/
MinimumBSTSumsubtree.cpp
File metadata and controls
39 lines (36 loc) · 1.21 KB
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
class Solution {
public:
int f(Node *root, int &minlen, int &sum, int target, bool &flag, int &maxi, int &mini)
{
if (root == NULL)
return 0;
// below subtrees sum,isbst,maxi,mini
int leftsum = 0, rightsum = 0;
bool leftflag = true, rightflag = true;
int leftmaxi = -1e9, rightmaxi = -1e9;
int leftmini = 1e9, rightmini = 1e9;
int leftlen = f(root->left, minlen, leftsum, target, leftflag, leftmaxi, leftmini);
int rightlen = f(root->right, minlen, rightsum, target, rightflag, rightmaxi, rightmini);
sum = root->data + leftsum + rightsum;
int currLen = leftlen + rightlen + 1;
maxi = max(root->data, max(leftmaxi, rightmaxi));
mini = min(root->data, min(leftmini, rightmini));
flag = false;
if (root->data > leftmaxi && root->data < rightmini && leftflag && rightflag)
{
flag = true; // current tree is also bst
}
if (currLen < minlen && sum == target && flag)
{
minlen = currLen;
}
return currLen;
}
int minSubtreeSumBST(int target, Node *root)
{
int sum = 0, minlen = 1e9, maxi = -1e9, mini = 1e9;
bool flag = true;
f(root, minlen, sum, target, flag, maxi, mini);
return minlen == 1e9 ? -1 : minlen;
}
};