-
Notifications
You must be signed in to change notification settings - Fork 617
/
spiral_traversal_of_binary_tree.cpp
140 lines (108 loc) · 2.72 KB
/
spiral_traversal_of_binary_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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
/*
We need to traverse the binary tree in spiral order. First row will be traversed from right to left,second from left to right and so on.
nullptr is pushed into the queue after every level. If we reach nullptr we will print the entire vector(arr) in reverse if flag==0 and forward if flag==1.
After printing we will make the flag reverse and resize the vector to 0. We will keep on doing this process and at the end when the queue becomes empty
we will get the spiral traversal printed.
*/
#include <bits/stdc++.h>
using namespace std;
class Node{
public:
int data;
Node* left;
Node* right;
// constructor
Node(int x){
data = x;
left = nullptr;
right = nullptr;
}
};
Node* Insert(Node* root, int data){
// make new Node
if(root == nullptr){
root = new Node(data);
return root;
}
else if(data < root->data){
root->left = Insert(root->left,data);
}
else{
root->right = Insert(root->right,data);
}
return root;
}
void spiral_level_order(Node * root) {
if (root == nullptr) return;
queue < Node * > q;
vector < int > arr;
int flag = 0;
q.push(root);
q.push(nullptr);
while (q.empty() == false) {
Node * temp = q.front();
q.pop();
//Printing level when nullptr arrived
if (temp == nullptr) {
//Printing in the reverse order
if (flag == 0) {
for (auto it = arr.end() - 1; it >= arr.begin(); it--) {
cout << * it << " ";
}
flag = 1;
}
//Printing in the forward order
else {
for (auto e: arr) {
cout << e << " ";
}
flag = 0;
}
arr.resize(0);
//To check we have covered entire tree or not
if (q.size() != 0) q.push(nullptr);
continue;
}
arr.push_back(temp -> data);
if (temp -> left != nullptr) q.push(temp -> left);
if (temp -> right != nullptr) q.push(temp -> right);
}
}
int main() {
int data;
cin >> data;
Node* root = new Node(data);
while(true){
cin >> data;
if(data==-1)break;
root = Insert(root, data);
}
spiral_level_order(root);
return 0;
}
/*
Input:
1 5 4 9 -1
Output:
1 5 9 4
Explanation:
1 <-
\
5 ->
/ \
4 9 <-
Input:
2 1 4 3 6 5 7 -1
Output:
2 1 4 6 3 5 7
Explanation:
2 <-
/ \
1 4 ->
/ \
3 6 <-
/ \
5 7 ->
Time Complexity: O(Nodes)
Space Complexity: O(Nodes)
*/