L2-006 树的遍历 (25 分)

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<bits/stdc++.h>
using namespace std;
queue<int> q;
vector<int> d[35];
int maxk=0;
void dfs(int a[],int al,int b[],int k)
{
    if(al==0){maxk=max(maxk,k); return ;}
    d[k].push_back(a[al]);
    int i;
    for(i=1;i<=al;i++) if(b[i]==a[al]) break;
    int len1=i-1,len2=al-i;
    dfs(a,len1,b,k+1);
    dfs(a+i-1,len2,b+i,k+1);
}
int main()
{
    int a[35],b[35];
    int n;
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int j=1;j<=n;++j) cin>>b[j];
    dfs(a,n,b,1);
    for(int i=1;i<maxk;++i){
        for(int j=0;j<d[i].size();++j){
            if(i==maxk-1&&j==d[i].size()-1) printf("%d\n",d[i][j]);
            else printf("%d ",d[i][j]);
        }
    }
    return 0;
}

收起测试区

测试用例

1

7
2 3 1 5 7 6 4
1 2 3 4 5 6 7