生活的烦恼
时间限制:
1000 ms | 内存限制:
65535 KB
难度:
2
- 描述
-
生活的暑假刚集训开始,他要决心学好字典树,二叉树,线段树和各种树,但生活在OJ上刷题的时候就遇到了一个特别烦恼的问题。那当然就是他最喜欢的二二叉树咯!题目是这样的:给你一颗非空的二叉树,然后再给你一个整数n,让生活输出这颗二叉树的第n(n>0且n<=树的深度)层,出题者为了给生活降低难度,要求两个输出数据之间用'~'隔开。看来我们的出题人很有爱啊!
- 输入
- 第一行输入一个数N,表示有N组测试数据。接下来N行,每行一个字符串,用'#'表示为空的节点,树的结束标志为'@'。'@'后仅有一个空格,空格后为一个数字,表示生活要输出的二叉树的第几层!
- 输出
- 每行输出一个字符串,表示给出二叉树的第n层!
- 样例输入
-
2 1 2 # # 3 # # @ 1 5 7 3 # # # 4 # # @ 3
- 样例输出
-
1 3
- 提示
-
5 第一层
/ \
7 4 第二层
/
3 第三层
-
-
一开始想用层序遍历处理,发现很麻烦处理这个层的关系,后来还是递归+队列处理了
-
-
#include<queue> #include<iostream> #include<malloc.h> using namespace std; struct tree { char c; tree *left, *right; }; int n; queue<char> rel; void CreatTree(tree *&t) {//建树 char d; cin >> d; if(d == '#') t = NULL; else { t = (tree*)malloc(sizeof(tree)); t -> c = d; CreatTree(t -> left); CreatTree(t -> right); } } void Print(tree *t, int level) {//level记录每个节点的层数 if(t != NULL) { if(level == n) rel.push(t -> c);//level = n就入队 Print(t -> left, level+1); Print(t -> right, level+1); } } int main() { char s; int t; cin >> t; while(t--) { tree *a = NULL; CreatTree(a); cin >> s >> n; Print(a, 1); if(!rel.empty()) { cout << rel.front(); rel.pop(); } while(!rel.empty()) { cout << "~" << rel.front(); rel.pop(); } cout << endl; } }