L2-006 树的遍历 （25 分）

 

 

链接

https://pintia.cn/problem-sets/994805046380707840/problems/994805069361299456

题目

给定一棵二叉树的后序遍历和中序遍历，请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。

输入格式：

输入第一行给出一个正整数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<cstring>
#include<cstdio>
#include<queue>
using namespace std;
const int maxn=50;
struct node{
    int data;
    node * lchild,* rchild;
};
int n;
int in[maxn];
int post[maxn];
node *creat(int postl,int postr,int inl,int inr){
    if(inl>inr) return NULL;
    node *root=new node;
    root->data=post[postr];
    int index=-1;
    for(int i=inl;i<=inr;i++){
        if(in[i]==post[postr]){
            index=i;
            break;
        }
    }
    int num=index-inl;
    root->lchild=creat(postl,postl+num-1,inl,index-1);
    root->rchild=creat(postl+num,postr-1,index+1,inr);
    return root;
}
void bfs(node *root){
    queue<node* > q;
    q.push(root);
    int cnt=0;
    while(!q.empty()){
        node* a=q.front();
        q.pop();
        if(cnt==0){
            printf("%d",a->data);cnt++;
        }else{
            printf(" %d",a->data);
        }
        if(a->lchild!=NULL) q.push(a->lchild);
        if(a->rchild!=NULL) q.push(a->rchild);
    }
}
int main(){
    cin>>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=creat(0,n-1,0,n-1);
    bfs(root);
    return 0;
}