序列构建二叉树 - 树的遍历 - 天梯赛 L2-006

序列构建二叉树 - 树的遍历 - 天梯赛 L2-006

题意：

$给定一个n个结点的二叉树，$

$根据后序遍历序列和中序遍历序列，输出层次遍历的序列。$

输入：

$首行一个正整数n，表示结点的数量$

$接着一行为树的后序遍历序列。$

$最后一行为树的中序遍历序列。$

输出：

$一行数据，表示层次遍历序列。$

输入样例：

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

输出样例：

4 1 6 3 5 7 2

数据范围：

$n\le 30$

时间限制: 400 ms

---

分析：

主要思考一下建数：

$无论是先序+中序，还是后序+中序，$

$我们首先要从先序/后序序列中，找到根结点u，然后在中序序列中找到u结点，将中序序列一分为二。$

$然后递归建立左右子树。$

有两个需要注意的地方：

$$\begin{gathered} ①、因为我们是递归对两个子区间[l,m]和[m+1,r]进行操作， \\ 边界情况当m是中序序列中的倒数第二个结点的时候，m+1=r，此时会返回空结点。 \\ 但事实上它是叶子结点，因此，我们建树的区间写成[1,n+1]。 \end{gathered}$$

$$\begin{gathered} ②、因为本题是后序+中序，后序的遍历顺序为：左右根，而我们是从后往前遍历后序序列来找根的， \\ 所以，反过来我们应该优先递归建立好右子树，再递归建立左子树。 \\ 否则会导致左右孩子错位。 \end{gathered}$$

层次遍历：

$非常容易地应当选择bfs来输出。$

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>

using namespace std;

const int N = 40;

int n;
int suf[N], mid[N];
struct node
{
    int l, r;
}tr[N*4];
int root;
int ans[N], idx;

int build(int l,int r)
{
    if(l>=r) return -1;
    int m;
    int u=suf[root--];
    for(int i=1;i<=n;i++) 
        if(mid[i]==u)
        {
            m=i;
            break;
        }
    tr[u].r=build(m+1,r);
    tr[u].l=build(l,m);
    return u;
}

void bfs(int S)
{
    queue<int>Q;
    Q.push(S);
    
    while(Q.size())
    {
        int u=Q.front();
        Q.pop();
        if(u==-1) continue ;
        ans[idx++] = u;
        Q.push(tr[u].l), Q.push(tr[u].r);
    }
}

int main()
{
    cin>>n;root=n;
    for(int i=1;i<=n;i++) cin>>suf[i];
    for(int i=1;i<=n;i++) cin>>mid[i];
    
    int start=build(1,n+1);
    bfs(start);
    
    for(int i=0;i<idx;i++) 
        printf("%d%c",ans[i],(i==idx-1?'\n':' '));

    return 0;
}