天梯赛l2-011玩转二叉树（层序遍历，25分代码）

l2-011

前言

这道题和l2-006树的遍历几乎一模一样，我会在思路中讲解中序加前序→层序

题目

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

输入格式：

输入第一行给出一个正整数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

思路

凡是这种什么遍历转什么遍历的题目的核心都在于找当前区间的根。

以该题为例，

设中序遍历序列为Z[N]，前序遍历序列为Q[N]；

设中序遍历左端点为ZL，右端点为ZR;设前序遍历左端点为QL，右端点为QR。

前序遍历的最左边一定是根，得到根的值后遍历对应的中序遍历得到中序遍历里面根的位置，设这个位置是i，那么根据中序遍历可以得出左子树的长度len=i-ZL;

方便起见，len=i-ZL-1;

于是：

中序遍历对应左子树是[ZL,i-1]，右子树是[i+1,ZR]；

前序遍历对应左子树是[QL + 1, QL + 1 + len]，右子树是[QL + 1 + len + 1, QR];

左子树的根的值是Q[QL + 1]；右子树的根的值是Q[QL + 1 + len + 1]。

准备工作弄完了。

接下来是层序遍历，层序遍历在于给同一层的节点编号，比如给根节点编号为bh,左子树根节点编号就得是2*bh，右子树编号就得是2*bh+1，后续按照编号顺序输出。

这道题要求镜面翻转后的层序遍历，那么要翻转树吗？

不用，只需要将层序遍历翻转就能满足题意，将左子树编号由2*bh改为2*bh+1，右子树编号由2*bh+1改为2*bh。

可以点击→l2-006查看l2-006。

AC代码

#include <iostream>
#include <vector>
#include <map>
#include <algorithm>
using namespace std;
const int N = 31;
int n, q[N], z[N];
map<int, int>mp;
void cxbl(int father, int ql, int qr, int zl, int zr, int bh)
{
	if (ql > qr || zl > zr)return;
	mp[bh] = father;
	int p = q[ql], i = zl;
	for (; i <= zr; ++i)if (z[i] == p)break;
	int len = i - zl - 1;//这里是长度减一，为了方便起见
	cxbl(q[ql + 1], ql + 1, ql + 1 + len, zl, i - 1, 2 * bh + 1);
	cxbl(q[ql + 1 + len + 1], ql + 1 + len + 1, qr, i + 1, zr, 2 * bh);
}
int main()
{
	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
	cin >> n;
	for (int i = 1; i <= n; ++i)cin >> z[i];
	for (int i = 1; i <= n; ++i)cin >> q[i];
	cxbl(q[1], 1, n, 1, n, 1);
	vector<int>num;
	for (auto x : mp)num.push_back(x.second);
	for (int i = 0; i < n - 1; ++i)cout << num[i] << " ";
	cout << num[n - 1];
	return 0;
}

千里之行，始于足下。