PAT L2-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>

typedef struct node
{
    int data;
    struct node *l,*r;
} BTnode, *BiTree;

int flag, n;

BiTree Rstore(int b[],int c[],int n)
{
    if(n==0)return NULL;
    BiTree bt;
    bt = (BTnode*)malloc(sizeof(BTnode));
    bt->data = c[n-1];
    int i;
    for( i = 0; i < n; i++ )
    {
        if(b[i]==c[n-1])break;
    }
    bt->l = Rstore(b,c,i);
    bt->r = Rstore(b+i+1,c+i,n-i-1);
    return bt;
}
void levelorder(BiTree bt)
{
    BiTree *q;
    q = (BiTree *)malloc(sizeof(BiTree)*n);
    int in,out;
    if(bt==NULL)return;
    in = 0;
    out = 0;
    q[in++] = bt;
    while(in>out)
    {
        if(q[out])
        {
            if(!flag)
            {
                printf("%d",q[out]->data);
                flag = 1;
            }
            else printf(" %d",q[out]->data);
            q[in++] = q[out]->l;
            q[in++] = q[out]->r;
        }
        out++;
    }
}
int main()
{
    BiTree T;
    int *b, *c;
    int i;
    scanf("%d",&n);
    b = (int *)malloc(sizeof(int)*n);
    c = (int *)malloc(sizeof(int)*n);
    for( i = 0; i < n; i++ )
        scanf("%d",&b[i]);
    for( i = 0; i < n; i++ )
        scanf("%d",&c[i]);
    flag = 0;
    T = Rstore(c,b,n);
    levelorder(T);
    printf("\n");
    return 0;
}