PTA L2-006 树的遍历 (25 分)_给定一棵二叉树的后序遍历和中序遍历,请你输出其层序遍历的序列。这里假设键值都-CSDN博客

本文链接：https://blog.csdn.net/m0_51864047/article/details/115194731

L2-006 树的遍历 (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<queue>
using namespace std;
const int maxn=35;
int post[maxn],mid[maxn],pos[maxn],n,p;
struct N{int data;N *l,*r;};

void creat(int l,int r,N* &fa){
    for(int i=l;i<=r;i++)   if(mid[i]==post[p]){fa=new N{post[p--]};break;}
    if(fa)  creat(pos[fa->data]+1,r,fa->r),creat(l,pos[fa->data]-1,fa->l);
}

int main(){
    cin>>n,p=n;
    for(int i=1;i<=n;i++)   cin>>post[i];
    for(int i=1;i<=n;i++)   cin>>mid[i],pos[mid[i]]=i;
    N *root;
    creat(1,n,root);
    queue<N> que;   que.push(*root);
    while(que.size()){
        N t=que.front();que.pop();
        if(t.l) que.push(*t.l),delete t.l;
        if(t.r) que.push(*t.r),delete t.r;
        cout<<t.data<<(que.size()?' ':'\n');
    }
}