玩转二叉树（二叉树）

L2-011 玩转二叉树 (pintia.cn)

题目描述

给定一棵二叉树的中序遍历和前序遍历，请你先将树做个镜面反转，再输出反转后的层序遍历的序列。所谓镜面反转，是指将所有非叶结点的左右孩子对换。这里假设键值都是互不相等的正整数。

输入格式：

输入第一行给出一个正整数N（≤30），是二叉树中结点的个数。第二行给出其中序遍历序列。第三行给出其前序遍历序列。数字间以空格分隔。

输出格式：

在一行中输出该树反转后的层序遍历的序列。数字间以1个空格分隔，行首尾不得有多余空格。

输入样例：

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

输出样例：

4 6 1 7 5 3 2

题目分析

本题为上次 树的遍历 的又一变形，考察为通过树的 中序遍历 和 前序遍历 求出树的 镜像层序遍历。

由于本题与上次思路多有重合之处，故采取另一种写法来进行解答。

关于树的中序遍历及前序遍历的相关性质不再赘述，本次解法我们为每一层开一个 vector 存储，针对每一层进行一次镜像层序遍历，具体操作为对当前层操作，即将对应节点压入所在的 vector 结束后，首先先进行右子树的深度搜索，再进行左子树的深度搜索，注意对当前节点所属层数的传递。

同样对于节点数最多为 $30$，我们需要考虑树的层数最多为 $30$ 层的限制，即将 vector 的行数设置为 $30$ 以上。

Accept代码

#include <bits/stdc++.h>

using namespace std;

const int N = 35;
vector<int> g[N];
int n;
int m[N], l[N];

void dfs(int l1, int r1, int l2, int r2, int x)
{
    if (l1 > r1 || l2 > r2) return ;
    int root = l[l2];
    g[x].push_back(root);
    int k;
    for (int i = l1; i <= r1; i ++)
        if (m[i] == root)
        {
            k = i - l1;
            break;
        }
    dfs(l1 + k + 1, r1, l2 + k + 1, r2, x + 1);
    dfs(l1, l1 + k - 1, l2 + 1, l2 + k, x + 1);
}

int main()
{
    cin >> n;
    for (int i = 0; i < n; i ++) cin >> m[i];
    for (int i = 0; i < n; i ++) cin >> l[i];
    
    dfs(0, n - 1, 0, n - 1, 0);
    cout << g[0][0];
    for (int i = 1; g[i].size(); i ++)
        for (int j = 0; j < g[i].size(); j ++)
            cout << ' ' << g[i][j];
    return 0;
}