NYOJ 1063 生活的烦恼

生活的烦恼

时间限制：1000 ms  |  内存限制：65535 KB

难度：2

描述

生活的暑假刚集训开始，他要决心学好字典树，二叉树，线段树和各种树，但生活在OJ上刷题的时候就遇到了一个特别烦恼的问题。那当然就是他最喜欢的二二叉树咯！题目是这样的：给你一颗非空的二叉树，然后再给你一个整数n，让生活输出这颗二叉树的第n(n>0且n<=树的深度)层，出题者为了给生活降低难度，要求两个输出数据之间用'~'隔开。看来我们的出题人很有爱啊！

输入

第一行输入一个数N，表示有N组测试数据。接下来N行，每行一个字符串，用'#'表示为空的节点，树的结束标志为'@'。'@'后仅有一个空格，空格后为一个数字，表示生活要输出的二叉树的第几层！

输出

每行输出一个字符串，表示给出二叉树的第n层！

样例输入

21 2 # # 3 # # @ 15 7 3 # # # 4 # # @ 3

样例输出

13

提示

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;  }}