L2-006. 树的遍历 -天梯赛练习题

L2-006. 树的遍历

  

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

输入格式：

输入第一行给出一个正整数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 <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <cmath>
#include <cstdlib>
#include <ctime>
#include <stack>
using namespace std;
const int maxn=50;
int mid[maxn],be[maxn];
struct node{
	int l,r;
}a[maxn];

int rebuild(int la,int ra,int lb,int rb){///a是中序遍历  b是后序遍历

    if(ra<la) return 0;///结束条件，很重要。

    int rt = be[rb];
    int i;
    ///在中序遍历中找到当前根的位置
    for(i = 0;;i++)
        if(mid[la+i]==rt) break;

    a[rt].l = rebuild(la,la+i-1,lb,lb+i-1);
    a[rt].r = rebuild(la+i+1,ra,lb+i,rb-1);
    return rt;///返回给上一层
}

void bfs(int s){
    queue<int> q;
    q.push(s);
    while(!q.empty()){
        int t = q.front();
        q.pop();
        if(a[t].l!=0) q.push(a[t].l);
        if(a[t].r!=0) q.push(a[t].r);
        printf("%d",t);
        if(!q.empty()) printf(" ");
    }
}

int main()
{
    int n;
    scanf("%d",&n);
    for(int i = 0; i<n; ++i){
        scanf("%d",&be[i]);
    }
    for(int i = 0; i<n; ++i){
        scanf("%d",&mid[i]);
    }
    rebuild(0,n-1,0,n-1);
    bfs(be[n-1]);
    return 0;
}