NYOJ 1063 生活的烦恼

生活的烦恼

时间限制: 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;
  }
}        


猜你喜欢

转载自blog.csdn.net/qq799028706/article/details/78747952