天梯

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

输入格式：

输入第一行给出一个正整数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>
const int INF=0x3f3f3f3f;
typedef long long LL;
const double eps =1e-8;
const int mod=1e9+7;
const int maxn=1e5+10;
using namespace std;

int n;
int a[50]; //后序 
int b[50]; //中序 
vector<int> vt; //存层序 
int T[50][2];

void solve(int L1,int R1,int L2,int R2,int pre,int f)//f表示左还是右 
{
    int rt=a[R1];
    if(pre!=-1)
        T[pre][f]=rt;
    int pos;
    for(int i=L2;i<=R2;i++)
    {
        if(b[i]==rt)
        {
            pos=i;
            break;
        }
    }
    int la=pos-L2;
    int lb=R2-pos;
    if(la) solve(L1,L1+la-1,L2,pos-1,rt,0);
    if(lb) solve(R1-lb,R1-1,pos+1,R2,rt,1);
}

void BFS(int st)//求出层序遍历 
{
    queue<int> qe;
    qe.push(st);
    while(!qe.empty())
    {
        int u=qe.front(); qe.pop();
        vt.push_back(u);
        for(int i=0;i<=1;i++)
        {
            int v=T[u][i];
            if(v) qe.push(v);
        }
    }
}

int main()
{
    #ifdef DEBUG
    freopen("sample.txt","r",stdin);
    #endif
    
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=1;i<=n;i++)
        scanf("%d",&b[i]);
    solve(1,n,1,n,-1, -1);
    BFS(a[n]);
    for(int i=0;i<n;i++)
        printf(i==n-1?"%d
":"%d ",vt[i]);
    
    return 0;
}

-