团体程序设计天梯赛PTA L2-006树的遍历

题意：已知中序遍历和后序遍历得出层序遍历

思路：先将中序遍历和后序遍历还原出树，在bfs打印出来。

后续遍历是左右根，所以最后一个是根，由根再到中序遍历中寻找左右子树的元素。利用递归的方法还原出树。树是用结构体数组保存，用链表应该也行。

#include <iostream>
#include<algorithm>
#include<cstdio>
#include<string>
#include<cstring>
#include<vector>
#include<queue>
using namespace std;
const int N = 50;
int hou[N];
int zhong[N];
struct node
{
    int l,r;
}a[N];
int n;
//la,ra是中序遍历 lb,rb是后序遍历
int creat(int la,int ra,int lb,int rb)
{
    if(la > ra)return 0;
    int root = hou[rb],p1,p2;
    p1 = la;
    while(zhong[p1]!=root)p1++;//找根节点
    p2 = p1-la;//p2是左子树的个数
    a[root].l=creat(la,p1-1,lb,lb+p2-1);
    a[root].r=creat(p1+1,ra,lb+p2,rb-1);
    return root;
}
void bfs()
{
    queue<int>q;
    vector<int>v;
    q.push(hou[n-1]);
    while(!q.empty())
    {
        int w = q.front();
        q.pop();
        if(w==0)
        {
            break;
        }
        v.push_back(w);
        if(a[w].l!=0)q.push(a[w].l);
        if(a[w].r!=0)q.push(a[w].r);
    }
    for(int i = 0; i < n; i ++)
    {
        if(i == 0)
        printf("%d",v[i]);
        else
            printf(" %d",v[i]);
    }
}
int main()
{
    scanf("%d",&n);
    for(int i = 0; i < n; i ++)
    {
        scanf("%d",&hou[i]);
    }
    for(int i = 0; i < n; i ++)
    {
        scanf("%d",&zhong[i]);
    }
    creat(0,n-1,0,n-1);
    bfs();
    return 0;
}