-
Notifications
You must be signed in to change notification settings - Fork 1
/
transform_sum_tree.cpp
103 lines (88 loc) · 2.05 KB
/
transform_sum_tree.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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
#include <bits/stdc++.h>
using namespace std;
struct Node
{
int data;
struct Node *left;
struct Node *right;
};
Node* newNode(int val)
{
Node* temp = new Node;
temp->data = val;
temp->left = NULL;
temp->right = NULL;
return temp;
}
Node* buildTree(string str)
{
if(str.length() == 0 || str[0] == 'N')
return NULL;
vector<string> ip;
istringstream iss(str);
for(string str; iss >> str; )
ip.push_back(str);
Node* root = newNode(stoi(ip[0]));
queue<Node*> queue;
queue.push(root);
int i = 1;
while(!queue.empty() && i < ip.size()) {
Node* currNode = queue.front();
queue.pop();
string currVal = ip[i];
if(currVal != "N") {
currNode->left = newNode(stoi(currVal));
queue.push(currNode->left);
}
i++;
if(i >= ip.size())
break;
currVal = ip[i];
if(currVal != "N") {
currNode->right = newNode(stoi(currVal));
queue.push(currNode->right);
}
i++;
}
return root;
}
void inorder(Node * node)
{
if(node==NULL)
return;
inorder(node->left);
cout<<node->data<<" ";
inorder(node->right);
}
class Solution {
public:
int traverseAndSum(Node* root) {
if (!root) return 0;
int lst = traverseAndSum(root->left);
int rst = traverseAndSum(root->right);
int tmp = root->data;
if (lst == 0 && rst == 0) root->data = 0;
else root->data = lst + rst;
return tmp + lst + rst;
}
void toSumTree(Node *node)
{
traverseAndSum(node);
}
};
int main()
{
int t;
scanf("%d ",&t);
while(t--)
{
string s;
getline(cin,s);
Node* root = buildTree(s);
Solution ob;
ob.toSumTree(root);
inorder(root);
cout<<endl;
}
return 1;
}