2-006. 树的遍历

L2-006. 树的遍历

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

给定一棵二叉树的后序遍历和中序遍历，请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。

输入格式：

输入第一行给出一个正整数N（<=30），是二叉树中结点的个数。第二行给出其后序遍历序列。第三行给出其中序遍历序列。数字间以空格分隔。

输出格式：

在一行中输出该树的层序遍历的序列。数字间以1个空格分隔，行首尾不得有多余空格。

输入样例：

7
2 3 1 5 7 6 4
1 2 3 4 5 6 7

输出样例：

4 1 6 3 5 7 2

#include <bits/stdc++.h>
using namespace std;
struct node
{
    int data;
    struct node *l, *r;
};
int in[33],la[33];
struct node* build(int *in, int *la, int n)
{
    struct node *t;
    for(int i = 0; i < n; i++)
    {
        if(la[n-1] == in[i])
        {
            t = new node;
            t->data = in[i];
            t->l = build(in, la, i);
            t->r = build(in+i+1, la+i, n -( i + 1));
            return t;
        }
    }
    return NULL;
}
void in1(struct node *root)
{
    if(root)
    {
        in1(root->l);
        printf("%d ",root->data);
        in1(root->r);
    }
}
void bfs(struct node *root)
{
    queue<node *>q;
    q.push(root);
    int flag = 0;
    while(!q.empty())
    {
        node *u = q.front();
        q.pop();
        if(flag)
            printf(" ");
            printf("%d",u->data);
        flag = 1;
        if(u->l != NULL)
            q.push(u->l);
        if(u->r != NULL)
            q.push(u->r);
    }

}
int main()
{
    int n;
    scanf("%d", &n);
    for(int i = 0; i < n; i++)
        scanf("%d", &la[i]);
     for(int i = 0; i < n; i++)
        scanf("%d", &in[i]);
        struct node *root = new node;
        root = build(in, la, n);
        bfs(root);
}