7-10 树的遍历 (25 分)

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

输入格式：

输入第一行给出一个正整数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<iostream>
#include<vector>
#include<queue>
using namespace std;

const int maxn = 1e4+10;

int after[maxn],in[maxn];
struct node{
    int l,r;
}tr[maxn];

int dfs(int la,int ra,int lb,int rb){
    if(la>ra)return 0;
    int p1  = la;
    int ver = after[rb];
    while(in[p1]!=ver)p1++;
    int k = p1-la;
    tr[ver].l = dfs(la,p1-1,lb,lb+k-1);
    tr[ver].r = dfs(p1+1,ra,lb+k,rb-1);
    return ver;
}
void bfs(int x){
    queue<int>q;
    vector<int>p;
    q.push(x);
    while(q.size()){
        int t = q.front();
        if(!t)break;
        q.pop();
        p.push_back(t);
        if(tr[t].l){
            q.push(tr[t].l);
        }
        if(tr[t].r)q.push(tr[t].r);
    }
    for(int i=0;i<p.size()-1;i++)cout<<p[i]<<" ";
    cout<<p[p.size()-1];
}
int main(){
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)cin>>after[i];

    for(int i=1;i<=n;i++)cin>>in[i];

    dfs(1,n,1,n);
    bfs(after[n]);
}