L2-011. 玩转二叉树 （树的层序遍历）

L2-011. 玩转二叉树

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

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

输入格式：

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

// 首先，利用中序和前序先把树构建出来，不要尝试用数组来存树，因为这棵树的结点可能有2^30个
// 然后，对树进行层序遍历。这个题目要求先把树镜像，再遍历，我们这里将层序遍历子节点入队的顺序颠倒一下便是镜像的层序遍历了。
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;
struct node {
	node* left;
	node* right;
	int value;
}*root;
vector<int> in(31);
vector<int> pre(31);

node* built(int pl, int pr, int il, int ir, node*  root)
{
	if (il > ir || pl > pr)
		return NULL;
	root = new node();
	int value = pre[pl];
	int index = distance(in.begin(), find(in.begin(), in.end(), value));
	int len = index - il;
	root->left = built(pl + 1, pl + len, il, index - 1, root->left);
	root->right = built(pl + len + 1, pr, index + 1, ir, root->right);
	root->value = value;
	return root;
}

void re_level(node* root)
{
	int flag = false;
	queue<node*> q;
	node* n;
	q.push(root);
	while (1)
	{
		if (q.empty())
			return;

		n = q.front();
		q.pop();
		if (!flag) {
			cout << n->value;
			flag = true;
		}
		else
			cout << " " << n->value;

		// 这里先入结点，再入左结点
		if (n->right != NULL)
			q.push(n->right);
		if (n->left != NULL)
			q.push(n->left);
	}
}

int main() 
{
	//freopen("data.txt", "r", stdin);
	int N;
	cin >> N;
	for (int i = 0; i < N; i++)
		cin >> in[i];
	for (int i = 0; i < N; i++)
		cin >> pre[i];
	root = built(0, N - 1, 0, N - 1, root);
	re_level(root);
	return 0;
}