给定一棵二叉树的后序遍历和中序遍历,请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。
输入格式:
输入第一行给出一个正整数N(≤),是二叉树中结点的个数。第二行给出其后序遍历序列。第三行给出其中序遍历序列。数字间以空格分隔。
输出格式:
在一行中输出该树的层序遍历的序列。数字间以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;
int a[35],b[35],n;
struct node
{
int la,lb,ra,rb;
node(int la,int ra,int lb,int rb):la(la),ra(ra),lb(lb),rb(rb){}
};
void bfs()
{
int flag=1;
queue <node> qu;
qu.push(node(1,n,1,n));
while(!qu.empty())
{
node t=qu.front();
qu.pop();
int la=t.la,lb=t.lb,ra=t.ra,rb=t.rb,root=a[ra];
if(flag)
{
cout<<root;
flag=0;
}
else cout<<" "<<root;
for(int i=lb;i<=rb;i++)
{
if(b[i]==root) //用根节点进行判断
{
int ln=i-lb;
int rn=rb-i;
if(ln>0) qu.push(node(la,la+ln-1,lb,i-1)); //左边
if(rn>0) qu.push(node(la+ln,ra-1,i+1,rb)); //右边
}
}
}
cout<<"\n";
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++) cin>>a[i];
for(int i=1;i<=n;i++) cin>>b[i];
bfs();
return 0;
}
来源:https://www.cnblogs.com/ww123/p/10290228.html