PAT(天梯赛练习题) -L2-011. 玩转二叉树

题目链接

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

输入格式：

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

经典面试题，用指针的读者可以百度镜像二叉树学一下。

这里提供一个邪道(?)做法。根据镜像的原则可知。镜像后的二叉树对他进行层次遍历就是对原二叉树 右→左 层次遍历。

所以不用镜像二叉树也能完成题目，毕竟题目只是要输出结果而已（笑）。

代码如下

#include<iostream>
#include<string>
#include<cstring>
#include<vector>
#include<map>
#include<algorithm>
#include<queue>
#include<set>
#include<cstdio>
#include<functional>
#include<iomanip>
#include<cmath>
#include<stack>
using namespace std;
const int maxn = int(1e3) + 100;
const int inf = 0x7fffffff;
typedef long long LL;
typedef unsigned long long ull;
struct trees_traversal {
	vector<int> preOrder, inOrder;
	int trees[3500], n, cnt;
	void dfs_pm(int rt, vector<int> pres, vector<int> mids) {//前+中还原
		if (pres.size() == 0)return;
		int  root;
		for (int i = 0; i < mids.size(); i++)
			if (mids[i] == pres[0]) {
				root = i;
				break;
			}
		vector<int> presl, presr, midsl, midsr;
		for (int i = 1; i <= root; i++)  presl.push_back(pres[i]);
		for (int i = root + 1; i < pres.size(); i++)  presr.push_back(pres[i]);
		for (int i = 0; i < root; i++) midsl.push_back(mids[i]);
		for (int i = root + 1; i < mids.size(); i++) midsr.push_back(mids[i]);
		trees[rt] = pres[0];
		dfs_pm(rt << 1, presl, midsl);
		dfs_pm(rt << 1 | 1, presr, midsr);
	}
	void bfs_levelOrder() {//按层次遍历输出
		queue<int> que;
		int f = 0, root = 1;
		que.push(root);
		while (que.size()) {
			if (!f) f = 1;
			else printf(" ");
			root = que.front();
			printf("%d", trees[root]);
			que.pop();
			if (trees[root << 1 | 1] != -1) que.push(root << 1 | 1);//先右子树
			if (trees[root << 1] != -1) que.push(root << 1);
		}
		printf("\n");
	}
	void init(int _n) {//初始化
		memset(trees, -1, sizeof(trees));
		n = _n, cnt = 0;
		preOrder.clear();
		inOrder.clear();
	}
}tra;
int main() {
	int n, x;
	scanf("%d", &n);
	tra.init(n);
	for (int i = 0; i < n; i++) {
		scanf("%d", &x);
		tra.inOrder.push_back(x);
	}
	for (int i = 0; i < n; i++) {
		scanf("%d", &x);
		tra.preOrder.push_back(x);
	}
	tra.dfs_pm(1, tra.preOrder, tra.inOrder);
	tra.bfs_levelOrder();
}