PAT L2-006 树的遍历

PAT-L2-006-树的遍历

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

输入格式：

输入第一行给出一个正整数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 <cstdio>
#include <algorithm>
#include <cstring>
#include <string>
#include <cstdlib>
#include <vector>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <bitset>
using namespace std;
vector<int>ed,in,level(10000,-1);
void post(int root,int start,int end,int index){
    if(start>end) return ;
    int i = start;
    while(i<end&&in[i]!=ed[root])i++;
    level[index] = ed[root];
    post(root-1-end+i,start,i-1,2*index+1);
    post(root-1,i+1,end,2*index+2);
}
int main(int argc, const char * argv[]) {
    int n;
    cin>>n;
    ed.resize(n);
    in.resize(n);
    for(int i=0;i<n;i++){
        cin>>ed[i];
    }
    for(int i=0;i<n;i++){
        cin>>in[i];
    }
    post(n-1,0,n-1,0);
    for(int i=0;i<level.size();i++){
        if(level[i]!=-1&&!i){
            printf("%d",level[i]);
        }else if(level[i]!=-1){
            printf(" %d",level[i]);
        }
    }
    return 0;
}