-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathboundaryTraversalofBinaryTree.cpp
More file actions
48 lines (43 loc) · 1.17 KB
/
boundaryTraversalofBinaryTree.cpp
File metadata and controls
48 lines (43 loc) · 1.17 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
bool isleaf(TreeNode<int>*root){
if(root->left==NULL&&root->right==NULL) return true;
else return false;
}
void leftBoundary(TreeNode<int> *root,vector<int>&res){
TreeNode<int>* cur = root->left;
while(cur){
if(!isleaf(cur)) res.push_back(cur->data);
if(cur->left) cur= cur->left;
else cur=cur->right;
}
}
void rightBoundary(TreeNode<int> *root,vector<int>&res){
TreeNode<int>* cur = root->right;
vector<int>tmp;
while(cur){
if(!isleaf(cur)) tmp.push_back(cur->data);
if(cur->right) cur= cur->right;
else cur=cur->left;
}
for(int i=tmp.size()-1;i>=0;i--){
res.push_back(tmp[i]);
}
}
void addleaf(TreeNode<int>*root,vector<int>&res){
if(isleaf(root)){
res.push_back(root->data);
return;
}
if(root->left) addleaf(root->left, res);
if(root->right) addleaf(root->right, res);
}
vector<int> traverseBoundary(TreeNode<int> *root)
{
// Write your code here.
vector<int> res;
if(root==NULL) return res;
if(!isleaf(root)) res.push_back(root->data);
leftBoundary(root,res);
addleaf(root,res);
rightBoundary(root,res);
return res;
}