-
Notifications
You must be signed in to change notification settings - Fork 0
/
1027.cpp
123 lines (116 loc) · 1.89 KB
/
1027.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
#include <cstdio>
#include <queue>
#include <stack>
using namespace std;
int in[35], post[35];
struct node{
int data;
node* lc;
node* rc;
};
node* create(int inL, int inR, int postL, int postR){
if(inL > inR){
return NULL;
}
node* root = new node;
root->data = post[postR];
root->lc = root->rc = NULL;
int pos;
for(int i = inL; i <= inR; i++){
if(in[i] == post[postR]){
pos = i;
break;
}
}
root->lc = create(inL, pos - 1, postL, postL + pos - inL - 1);
root->rc = create(pos + 1, inR, postL + pos - inL, postR - 1);
return root;
}
int cnt = 0;
int n;
void zig(node* root){
queue<node*> q;
stack<node*> st;
stack<node*> st2;
st.push(root);
int flag = 0;
while(!q.empty() || !st.empty() || !st2.empty()){
while(!st.empty()){
node* top = st.top();
st.pop();
cnt++;
if(flag == 0){
if(top->lc != NULL){
q.push(top->lc);
}
if(top->rc != NULL){
q.push(top->rc);
}
}
else{
if(top->rc != NULL){
st2.push(top->rc);
}
if(top->lc != NULL){
st2.push(top->lc);
}
}
printf("%d", top->data);
if(cnt < n){
printf(" ");
}
else{
printf("\n");
}
flag = 1;
}
while(!q.empty()){
node* top = q.front();
q.pop();
cnt++;
if(top->lc != NULL){
st.push(top->lc);
}
if(top->rc != NULL){
st.push(top->rc);
}
printf("%d", top->data);
if(cnt < n){
printf(" ");
}
else{
printf("\n");
}
}
while(!st2.empty()){
node* top = st2.top();
st2.pop();
cnt++;
if(top->lc != NULL){
st.push(top->lc);
}
if(top->rc != NULL){
st.push(top->rc);
}
printf("%d", top->data);
if(cnt < n){
printf(" ");
}
else{
printf("\n");
}
}
}
}
int main(){
scanf("%d", &n);
for(int i = 0; i < n; i++){
scanf("%d", &in[i]);
}
for(int i = 0; i < n; i++){
scanf("%d", &post[i]);
}
node* r = create(0, n - 1, 0, n - 1);
zig(r);
return 0;
}