1、题目大意
- 题目链接UVa 122
- 给出二叉树的各个结点,建立二叉树,然后按层次遍历输出各个节点的值。
- 每个结点的描述为 (n, s) 。n 表示结点值,为正整数。s 为一个 L 和 R 组成的字符串,表示从树根到该结点的路径,L 表示左子树,R 表示右子树。每棵树的结点输入以 () 作为结束,整个输入以 EOF 作为结束。
- 如果生成的二叉树不完整,或者树中的某个结点重复了,则输出 not complete,否则按层次遍历输出各个节点的值。
2、解题思路
- 把二叉树存数组中的方法不可取,因为结点数最多256个,若所有结点形成一条链,则树高为256,数组开不下。只能选取链式结构存储二叉树。
- 二叉树结点的结构:结点值(int);左子树,右子树(指针);是否有值(bool)。
- 设置 bool 型全局变量 is_tree = true。
- 读入结点。以字符串形式读入,然后从字符串中剥离出 n 和 s。
- 把每个结点放在树上。根据 s ,找到二叉树上对应的结点,如果该结点没有值,将结点值赋为 n ,如果该结点已经有值,说明该结点重复了,令全局变量 is_tree = false。在找对应结点的路径上,如果某个结点不存在,则要建立该结点。
- 如果 is_tree = false , 输出 not complete,结束。
- 层次遍历。需要用到队列。根结点入队,当队列不空时,出队一个结点,将该结点的左子树入队,将该结点的右子树入队。如果该结点有值,则输出结点值;否则二叉树不完整,输出 not complete,结束。
3、参考代码
#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<queue>
#include<cctype>
using namespace std;
bool is_tree = true;
vector <int> ans;
struct node {
bool have_value;
int value;
node* left,* right;
node():have_value(false),value(0),left(NULL),right(NULL){}
};
void get_node(string& str, int& n, string& s) {
n = 0;
for (int i = 1; i < str.length(); i++) {
if (isdigit(str[i])) n = 10 * n + str[i] - '0';
if (isalpha(str[i])) s.push_back(str[i]);
}
}
void add_node(node* root, int& n, string& s) {
char c;
node* N = root;
for (int i = 0; i < s.length(); i++) {
c = s[i];
if (c == 'L') {
if (N->left == NULL) N->left = new node();
N = N->left;
}
else if (c == 'R') {
if (N->right == NULL) N->right = new node();
N = N->right;
}
}
if (N->have_value) is_tree = false;
else { N->value = n; N->have_value = true; }
}
bool is_complete(node* root) {
if (is_tree) {
queue <node*> q;
q.push(root);
while (!q.empty()) {
node* N = q.front(); q.pop();
if(N->left) q.push(N->left);
if(N->right) q.push(N->right);
if (N->have_value) ans.push_back(N->value);
else return false;
}
}
else return false;
return true;
}
void del_tree(node* N) {
if (N == NULL) return;
del_tree(N->left);
del_tree(N->right);
delete N;
}
int main() {
string str;
node* root = new node();
while (true) {
if (!(cin >> str)) break;
do {
if (strcmp(&str[0], "()") == 0) {
if (is_complete(root)) {
cout << ans[0];
for (int i = 1; i < ans.size(); i++)
cout << ' ' << ans[i];
cout << endl;
}
else cout << "not complete" << endl;
ans.clear(); del_tree(root); root = new node(); is_tree = true;
break;
}
int n; string s;
get_node(str, n, s);
add_node(root, n, s);
} while (cin >> str);
}
return 0;
}
4、解题感悟
- 第一次编写跟树相关的算法,发现要注意好多细节,而且调试难度也加大了。这道题搞了一下午,头疼。
- 程序动态申请内存,要注意内存泄露,应该在程序中及时释放不用的动态内存。
- 刘大爷书上还介绍了用数组存储二叉树的方式,有时间应该了解一下。