天梯赛：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

 思路：用后序和中序求前序遍历时每一层加上深度，用一个vector的二维数组记录每一层的数据，如第一层就是v[1]，第二层就是v[2]，因为前序遍历时每一层都会是从左到右被存入数组,所以最后二维数组的输出顺序是没有问题的。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<map>
#include<queue>
using namespace std;
int ans;
int last;
int sum=0;
const int N=1010;
int hou[32];
int in[32];
int n;
vector<int> v[N];
int cnt;
void dfs(int l1,int r1,int l2,int r2,int u)
{

    if(l2>r2||l1>r1)
        return;
    if(u>cnt)
        cnt=u;
        int c=hou[r1];
        // cout<<c<<' ';
         v[u].push_back(c);
        int k=-1;
        int k2;
        for(int i=l2;i<=r2;i++)
        {
            if(in[i]==c)
            {
            k=i-l2;
            k2=r2-i;
            break;
            }
        }
         dfs(l1,l1+k-1,l2,l2+k-1,u+1);
         dfs(l1+k,r1-1,l2+k+1,r2,u+1);
}
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
        cin>>hou[i];

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

    dfs(1,n,1,n,1);
for(int i=1;i<=cnt;i++)
    for(int j=0;j<v[i].size();j++)
    {
        if(i==cnt&&j==v[i].size()-1)
        cout<<v[i][j];
        else
    cout<<v[i][j]<<' ';
    }
    return 0;
}