天梯 L2 树的遍历（已知后序中序求层序）

树的遍历 （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<bits/stdc++.h>
#define ll long long
#define eps 1e-8
using namespace std;
const int N = 1e5+;
const int INF = 0x3f3f3f3f;
int n,hou[],in[];
vector <int> ans(,-);
void work(int root,int s,int e,int index){
    if (s>e) return;
    int i=s;
    while(i<e && in[i]!=hou[root]) i++;
    ans[index] = hou[root];
    work(root--e+i,s,i-,index*+);
    work(root-,i+,e,index*+);
}
int main(){
    scanf("%d",&n);
    for (int i=;i<n;i++) scanf("%d",&hou[i]);
    for (int i=;i<n;i++) scanf("%d",&in[i]);
    work(n-,,n-,);
    int cnt = ;
    for(int i=;i<ans.size();i++){
        if(ans[i]!=- && cnt!=n-) {
            printf("%d ",ans[i]);
            cnt++;
        }else if (ans[i]!=-){
            printf("%d\n",ans[i]);
            break;
        }
    }
    return ;
}