NYOJ1063 生活的烦恼(二叉树的建立)

描述

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

输入

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

输出

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

样例输入

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

样例输出

13

提示

5 第一层/ \7 4 第二层/3 第三层

思路

首先，这个题的题意描述有问题，不得不吐槽一下。。。

题目首先给了n组数据，然后给了一个二叉树的先序遍历，让你求一下第n层。

先序遍历的过程中，用#来表示一个空字符，@代表一个二叉树描述完毕，后面的数字为需要输出第几层。

我们按照先序遍历的次序，递归的建立这个二叉树，然后用一个队列来找到要求的那一层的所有元素，并且输出

代码

#include <cstdio>#include <cstring>#include <string>#include <set>#include <cmath>#include <iostream>#include <stack>#include <queue>#include <vector>#include <algorithm>#define mem(a,b) memset(a,b,sizeof(a))#define inf 0x3f3f3f3f#define debug() puts("what the fuck!!!")#define ll long longusing namespace std;typedef struct Bi_tree{    string data;    Bi_tree *lchild,*rchild;} Binode,*Bitree;void build(Bitree &r)//建树{    string temp;    cin>>temp;    if(temp=="#")        r=NULL;    else    {        r=new Binode;        r->data=temp;        build(r->lchild);        build(r->rchild);    }}queue<string>ans;void solve(Bitree &r,int step,int deep){    if(r==NULL)        return;    if(step==deep)    {        ans.push(r->data);        return;    }    solve(r->lchild,step+1,deep);    solve(r->rchild,step+1,deep);}int main(){    int t;    cin>>t;    while(t--)    {        string c;        int n;        Bitree T;        build(T);        cin>>c>>n;        solve(T,1,n);        int len=ans.size();        if(len>1)        {            for(int i=0; i<len-1; i++)            {                cout<<ans.front()<<"~";                ans.pop();            }        }        cout<<ans.front()<<endl;        ans.pop();    }    return 0;}