-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathcommonNodeinTwoBinaryTree.cpp
More file actions
41 lines (41 loc) · 1.01 KB
/
commonNodeinTwoBinaryTree.cpp
File metadata and controls
41 lines (41 loc) · 1.01 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
#include<iostream>
#include<stack>
#include<unordered_set>
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
void traverseLeft(TreeNode *node, std::stack<TreeNode*> &st1){
while(node){
st1.push(node);
node = node->left;
}
}
int solve(TreeNode* A, TreeNode* B){
//do it iteratively
int sum = 0;
std::stack<TreeNode*> st1,st2;
traverseLeft(A,st1);
traverseLeft(B,st2);
while(!st1.empty() && !st2.empty()){
TreeNode* node1 = st1.top(),*node2 = st2.top();
if(node1->val==node2->val){
sum+=node1->val;
st1.pop();
st2.pop();
traverseLeft(node1->right,st1);
traverseLeft(node2->right,st2);
}
else if(node1->val<node2->val){
st1.pop();
traverseLeft(node1->right,st1);
}
else{
st2.pop();
traverseLeft(node2->right,st2);
}
}
return sum;
}