-
Notifications
You must be signed in to change notification settings - Fork 3
Expand file tree
/
Copy pathmax_tree.cpp
More file actions
72 lines (64 loc) · 1.16 KB
/
max_tree.cpp
File metadata and controls
72 lines (64 loc) · 1.16 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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
#include <iostream>
#include <vector>
#include <stack>
#include <limits>
using namespace std;
class TreeNode {
public:
int val;
TreeNode *left, *right;
TreeNode(int val) {
this->val = val;
this->left = this->right = NULL;
}
};
TreeNode* maxTree(vector<int> A) {
stack<TreeNode*> s;
TreeNode* root = NULL;
for(int i = 0; i <= A.size(); i++)
{
int rightValue = (i == A.size())? numeric_limits<int>::max() :
A[i];
TreeNode * rightNode = new TreeNode (rightValue);
while(!s.empty())
{
TreeNode * currentNode = s.top();
if(rightNode->val < currentNode->val)
{
break;
}
s.pop();
if(s.empty())
{
rightNode->left = currentNode;
}
else
{
TreeNode * leftNode = s.top();
if(leftNode->val < rightNode->val)
{
leftNode->right = currentNode;
}
else
{
rightNode->left = currentNode;
}
}
}
s.push(rightNode);
}
return (s.top())->left;
}
int main(void)
{
vector<int> a;
a.push_back(2);
a.push_back(5);
a.push_back(6);
a.push_back(0);
a.push_back(3);
a.push_back(1);
TreeNode * n = maxTree(a);
//inorder(n);
return 0;
}