7-3 玩转二叉树

给定一棵二叉树的中序遍历和前序遍历，请你先将树做个镜面反转，再输出反转后的层序遍历的序列。所谓镜面反转，是指将所有非叶结点的左右孩子对换。这里假设键值都是互不相等的正整数。

输入格式：

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

输出格式：

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

输入样例：

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

输出样例：

4 6 1 7 5 3 2

#include<bits/stdc++.h>
using namespace std;
vector<int>D[35];
int q[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 ls=i-b1,rs=b2-i;
    if(rs)dfs(q[a2-rs+1],a2-rs+1,a2,i+1,b2,d+1);
    if(ls)dfs(q[a1+1],a1+1,a1+ls,b1,i-1,d+1);
}
int main()
{
    cin>>n;
    for(int i=0;i<n;i++) cin>>z[i];
    for(int i=0;i<n;i++) cin>>q[i];
    dfs(q[0],0,n-1,0,n-1,0);
    for(int i=0;i<=md;i++)
        for(int x:D[i])
            cout<<(f?" ":"")<<x,f=1;
}