7-4 树的遍历

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

输入格式：

输入第一行给出一个正整数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;
vector<int>D[35];
int h[35],z[35],f=0,n,md;
void dfs(int rt,int a1,int a2,int b1,int b2,int d)
{
    if(a1>a2) return;
    md=max(d,md);
    D[d].push_back(rt);
    if(a1==a2) return;
    int i=find(z+b1,z+b2,rt)-z;
    int l=i-b1,r=b2-i;
    if(l)dfs(h[a1+l-1],a1,a1+l-1,b1,i-1,d+1);
    if(r)dfs(h[a2-1],a2-r,a2-1,i+1,b2,d+1);
}
int main()
{
    cin>>n;
    for(int i=0;i<n;i++) cin>>h[i];
    for(int i=0;i<n;i++) cin>>z[i];
    dfs(h[n-1],0,n-1,0,n-1,0);
    for(int i=0;i<=md;i++)
        for(int x:D[i])
            cout<<(f?" ":"")<<x,f=1;
}