-
Notifications
You must be signed in to change notification settings - Fork 0
/
trees_on_the_level_uva122b.cpp
106 lines (94 loc) · 1.97 KB
/
trees_on_the_level_uva122b.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
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<vector>
#include<queue>
using namespace std;
const int maxn = 256 + 10;
struct Node{
bool have_value;
int v;
Node* left, *right;
Node():have_value(false),left(NULL),right(NULL){}
};
Node* root;
queue<Node*> freenodes; // 空闲列表
Node node[maxn]; // 内存池
void init() {
for(int i = 0; i < maxn; i++)
freenodes.push(&node[i]);
}
Node* newnode() {
Node* u = freenodes.front();
u->left = u->right = NULL; u->have_value = false; // 重新初始化该结点
freenodes.pop();
return u;
}
void deletenode(Node* u) {
freenodes.push(u);
}
bool failed;
void addnode(int v, char* s) {
int n = strlen(s);
Node* u = root;
for(int i = 0; i < n; i++)
if(s[i] == 'L') {
if(u->left == NULL) u->left = newnode();
u = u->left;
} else if(s[i] == 'R') {
if(u->right == NULL) u->right = newnode();
u = u->right;
}
if(u->have_value) failed = true;
u->v = v;
u->have_value = true;
}
void remove_tree(Node* u) {
if(u == NULL) return;
remove_tree(u->left);
remove_tree(u->right);
deletenode(u);
}
char s[maxn];
bool read_input() {
failed = false;
remove_tree(root);
root = newnode();
for(;;) {
if(scanf("%s", s) != 1) return false;
if(!strcmp(s, "()")) break;
int v;
sscanf(&s[1], "%d", &v);
addnode(v, strchr(s, ',')+1);
}
return true;
}
bool bfs(vector<int>& ans) {
queue<Node*> q;
ans.clear();
q.push(root);
while(!q.empty()) {
Node* u = q.front(); q.pop();
if(!u->have_value) return false;
ans.push_back(u->v);
if(u->left != NULL) q.push(u->left);
if(u->right != NULL) q.push(u->right);
}
return true;
}
int main() {
vector<int> ans;
init();
while(read_input()) {
if(!bfs(ans)) failed = 1;
if(failed) printf("not complete\n");
else {
for(int i = 0; i < ans.size(); i++) {
if(i != 0) printf(" ");
printf("%d", ans[i]);
}
printf("\n");
}
}
return 0;
}