forked from ishikalohia/algorithm_
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathzigzagtree
More file actions
99 lines (88 loc) · 2.39 KB
/
zigzagtree
File metadata and controls
99 lines (88 loc) · 2.39 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
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
#include<stack>
void zigZagOrder(BinaryTreeNode<int> *root) {
// Write your code here
stack<BinaryTreeNode<int> *> s1;
stack<BinaryTreeNode<int> *> s2;
s1.push(root);
while(!s1.empty()||!s2.empty()){
BinaryTreeNode<int>*current;
while(!s1.empty()){
current=s1.top();
if(current)
cout<<current->data<<" ";
s1.pop();
if(current->left)
s2.push(current->left);
if(current->right)
s2.push(current->right);
if(s1.empty()){
cout<<endl;
}
}
while(!s2.empty()){
current=s2.top();
if(current)
cout<<current->data<<" ";
s2.pop();
if(current->right)
s1.push(current->right);
if(current->left)
s1.push(current->left);
if(s2.empty()){
cout<<endl;
}
}
}
}
#include <iostream>
#include <queue>
#include <stack>
template <typename T>
class BinaryTreeNode {
public :
T data;
BinaryTreeNode<T> *left;
BinaryTreeNode<T> *right;
BinaryTreeNode(T data) {
this -> data = data;
left = NULL;
right = NULL;
}
};
using namespace std;
#include "solution.h"
BinaryTreeNode<int>* takeInput() {
int rootData;
//cout << "Enter root data : ";
cin >> rootData;
if(rootData == -1) {
return NULL;
}
BinaryTreeNode<int> *root = new BinaryTreeNode<int>(rootData);
queue<BinaryTreeNode<int>*> q;
q.push(root);
while(!q.empty()) {
BinaryTreeNode<int> *currentNode = q.front();
q.pop();
int leftChild, rightChild;
//cout << "Enter left child of " << currentNode -> data << " : ";
cin >> leftChild;
if(leftChild != -1) {
BinaryTreeNode<int>* leftNode = new BinaryTreeNode<int>(leftChild);
currentNode -> left =leftNode;
q.push(leftNode);
}
//cout << "Enter right child of " << currentNode -> data << " : ";
cin >> rightChild;
if(rightChild != -1) {
BinaryTreeNode<int>* rightNode = new BinaryTreeNode<int>(rightChild);
currentNode -> right =rightNode;
q.push(rightNode);
}
}
return root;
}
int main() {
BinaryTreeNode<int>* root = takeInput();
zigZagOrder(root);
}