给定一棵二叉树的后序遍历和中序遍历,请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。
输入格式:
输入第一行给出一个正整数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<queue>
#include<vector>
using namespace std;
struct ty
{
int l=0,r=0;
};
ty tree[44];
int n;
int a[44];
int b[44];
vector<int>ans;
int build(int l1,int r1,int l2,int r2)
{
if(l1>r1)return 0;
int tr=a[r1];
int p1=l2;
while(b[p1]!=tr)++p1;
int p2=p1-l2;
tree[tr].l=build(l1,l1+p2-1,l2,p1-1);
tree[tr].r=build(l1+p2,r1-1,p1+1,r2);
return tr;
}
void bfs(int x)
{
queue<int>q;
q.push(x);
while(!q.empty())
{
int xx=q.front();
q.pop();
ans.push_back(xx);
if(tree[xx].l!=0)q.push(tree[xx].l);
if(tree[xx].r!=0)q.push(tree[xx].r);
}
for(int i=0;i<ans.size()-1;++i)
{
cout<<ans[i]<<" ";
}
cout<<ans[ans.size()-1];
}
int main()
{
cin>>n;
for(int i=1;i<=n;++i)cin>>a[i];
for(int i=1;i<=n;++i)cin>>b[i];
int tr=build(1,n,1,n);
bfs(a[n]);
return 0;
}