题目描述

分析:
给出一棵树的后序遍历和中序遍历,输出一棵树的层序遍历。后序遍历的最后一个结点是根节点,先在中序遍历中找到根结点的位置,完后递归建树。输出的时候BFS遍历输出即为层序遍历结果。
#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
struct node{
int data;
node* lchild;
node* rchild;
};
int pre[55],in[55],post[55];
int n;
node* create(int postL,int postR,int inL,int inR){
if(postL>postR){
return NULL;
}
node* root=new node;
root->data=post[postR];
int k;
for(k=inL;k<=inR;k++){
if(in[k]==post[postR])
break;
}
int numLeft=k-inL;
root->lchild=create(postL,postL+numLeft-1,inL,k-1);
root->rchild=create(postL+numLeft,postR-1,k+1,inR);
return root;
}
int num=0;
void BFS(node* root){
queue<node*>q;
q.push(root);
while(!q.empty()){
node* now=q.front();
q.pop();
printf("%d",now->data);
num++;
if(num<n) printf(" ");
if(now->lchild!=NULL) q.push(now->lchild);
if(now->rchild!=NULL) q.push(now->rchild);
}
}
int main(){
scanf("%d",&n);
for(int i=0;i<n;i++){
scanf("%d",&post[i]);
}
for(int i=0;i<n;i++)
{
scanf("%d",&in[i]);
}
node*root=create(0,n-1,0,n-1);
BFS(root);
return 0;
}