PAT.A1020 Tree Traversals

​​返回目录​​

题意

给出一颗树的后序遍历和中序遍历的序列，求层次遍历序列。

样例（可复制）

7
2 3 1 5 7 6 4
1 2 3 4 5 6 7

//output
4 1 6 3 5 7 2

注意点

1. 几个遍历的顺序。先序遍历：中–左--右。中序遍历：左–中--右。后序遍历：左–右--中。
2. 建议读者手动模拟一遍create函数建立样例给出的树

#include<bits/stdc++.h>
using namespace std;

struct  node{
    int data;
    node* lc;
    node* rc;
};
const int maxn=50;
int per[maxn],post[maxn],in[maxn],n,num=0;
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->lc=create(postl,postl+numleft-1,inl,k-1);
    root->rc=create(postl+numleft,postr-1,k+1,inr);
    return root;
}
void BFS(node* root){
    queue<node*> q;//层次遍历使用queue
    q.push(root);
    while(!q.empty()){
        node* now=q.front();
        q.pop();
        printf("%d",now->data);
        num++;
        if(num<n)printf(" ");
        if(now->lc!=NULL)q.push(now->lc);
        if(now->rc!=NULL)q.push(now->rc);
    }
}
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=create(0,n-1,0,n-1);
    BFS(root);
    return 0;
}