天梯赛 L2-011 玩转二叉树 (25 分)

L2-011 玩转二叉树 (25 分)

思路：
通过前序遍历和中序遍历构建出二叉树，再将树通过递归镜面翻转，最后层序遍历输出

预备知识：前序+中序遍历构建二叉树

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

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

c++

using namespace std;
#include <iostream>
#include<vector>
#include<queue>
#include<map>

#define element int
struct TreeNode {
	element val;
	TreeNode* left;
	TreeNode* right;
	TreeNode() : val(0), left(nullptr), right(nullptr) {}
	TreeNode(element x) : val(x), left(nullptr), right(nullptr) {}
	TreeNode(element x, TreeNode* left, TreeNode* right) : val(x), left(left), right(right) {}
};
map<element, int>m;
TreeNode* building(const vector<element>& preorder, const vector<element>& inorder, int pre_l, int pre_r, int ior_l, int ior_r)
{
	if (pre_l > pre_r)return NULL;

	int pre_root = pre_l;//根节点在前序遍历数组中的下标
	int ior_root = m[preorder[pre_root]];//根节点在中序遍历数组中的下标
	int L_size = ior_root - ior_l;//根节点左子树的大小

	TreeNode* root = new TreeNode(preorder[pre_root]);//创建根节点

	root->left = building(preorder, inorder, pre_l + 1, pre_l + L_size, ior_l, ior_root - 1);
	root->right = building(preorder, inorder, pre_l + L_size + 1, pre_r, ior_root + 1, ior_r);

	return root;
}
void mirror(TreeNode* T)//翻转二叉树
{
	swap(T->left , T->right);
	if (T->left)mirror(T->left);
	if (T->right)mirror(T->right);
}
void speak(TreeNode* t)
{
	queue<TreeNode*> q;
	int k = 0;
	if (t)q.push(t);
	while (q.size())
	{
		int n = q.size();
		for (int i = 0; i < n; i++)
		{
			if (!k++)cout << q.front()->val;
			else cout << " " << q.front()->val;
			if (q.front()->left)q.push(q.front()->left);
			if (q.front()->right)q.push(q.front()->right);
			q.pop();
		}
	}
}

int main()
{
	int n; cin >> n;
    vector<element>pre(n, 0), ior(n, 0);
    for (int i = 0; i < n; i++)cin >> ior[i];
    for (int i = 0; i < n; i++)cin >> pre[i];
    for (int i = 0; i < n; i++)m[ior[i]] = i;
	TreeNode* root = building(pre, ior, 0, pre.size() - 1, 0, ior.size() - 1);
	mirror(root);
	speak(root);
}