【天梯赛：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

思路

树的遍历，后序遍历最后一个是根节点，在中序遍历中找到根节点划分左右子树，然后递归建树。层序遍历就是bfs用队列实现

后序遍历和中序遍历结果数组，节点结构体

int last[55], mid[55];
struct nod{
    nod* l = NULL, *r = NULL;
    int val;
};

建树

nod* build(int l1, int r1, int l2, int r2)
{
    if(l1>r1 || l2>r2) return NULL;
    nod*node = new nod;
    node->val = last[r1];

    int x = -1;
    for(int i=l2; i<=r2; i++){
        if(mid[i] == last[r1]){
            x = i;
            break;
        }
    }
    x = x-l2;
    
    node->l = build(l1, l1+x-1, l2, l2+x-1);
    node->r = build(l1+x, r1-1, l2+x+1, r2);
    return node;
}

层序遍历

void bfs(nod *node)
{
    queue<nod*>q;
    if(node){
        q.push(node);
    }

    while(q.size()){
        nod *tem = q.front();
        q.pop();
        cout<<tem->val;
        if(tem->l) q.push(tem->l);
        if(tem->r) q.push(tem->r);
        if(q.size()) cout<<" ";
   }
}

完整代码

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

int last[55], mid[55];
struct nod{
    nod* l = NULL, *r = NULL;
    int val;
};

nod* build(int l1, int r1, int l2, int r2)
{
    if(l1>r1 || l2>r2) return NULL;
    nod*node = new nod;
    node->val = last[r1];

    int x = -1;
    for(int i=l2; i<=r2; i++){
        if(mid[i] == last[r1]){
            x = i;
            break;
        }
    }
    x = x-l2;
    
    node->l = build(l1, l1+x-1, l2, l2+x-1);
    node->r = build(l1+x, r1-1, l2+x+1, r2);
    return node;
}

void bfs(nod *node)
{
    queue<nod*>q;
    if(node){
        q.push(node);
    }

    while(q.size()){
        nod *tem = q.front();
        q.pop();
        cout<<tem->val;
        if(tem->l) q.push(tem->l);
        if(tem->r) q.push(tem->r);
        if(q.size()) cout<<" ";
   }
}

int main()
{
    ios::sync_with_stdio(0), cin.tie(0),cout.tie(0);
    int n; cin >> n;
    for(int i=1;i<=n;i++) cin >> last[i];
    for(int i=1;i<=n;i++) cin >> mid[i];

    nod* node = build(1,n,1,n);
    bfs(node);
}