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