天梯赛 L2-006 树的遍历 (25 分)

L2-006 树的遍历 (25 分)

思路：
利用中序遍历和后序遍历构建出二叉树，再层序遍历输出。

给定一棵二叉树的后序遍历和中序遍历，请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。

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

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

输入样例：
7
2 3 1 5 7 6 4
1 2 3 4 5 6 7
输出样例：
4 1 6 3 5 7 2

预备知识：根据后序和中序遍历构建出二叉树 我终于用上自己写的文章了，好感动T_T

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* build(const vector<element>& ior, const vector<element>& fin, int ior_l, int ior_r, int fin_l, int fin_r)
{
	if (fin_l > fin_r)return NULL;
	int fin_root = fin_r;//根节点在后序遍历数组中的下标
	int ior_root = m[fin[fin_root]];//根节点在中序遍历数组中的下标
	int L_size = ior_root - ior_l;//根节点左子树的大小
	TreeNode* root = new TreeNode(fin[fin_root]);//创建根节点
	root->left = build(ior, fin, ior_l, ior_root - 1, fin_l, fin_l + L_size - 1);
	root->right = build(ior, fin, ior_root + 1, ior_r, fin_l + L_size, fin_root - 1);
	return root;

}

void travel(TreeNode* T)//层序遍历
{
	int n;
	queue<TreeNode*>q;
	q.push(T); bool first = false;
	while (q.size())
	{
		n = q.size();
		for (int i = 0; i < n; i++)
		{
			if (first)cout << " ";
			cout << q.front()->val;
			if (q.front()->left)q.push(q.front()->left);
			if (q.front()->right)q.push(q.front()->right);
			q.pop();
			first = 1;
		}
	}
}

int main()
{
	int n; cin >> n;
	vector<int>ior(n, 0), fin(n, 0);
	for (int i = 0; i < n; i++)cin >> fin[i];//输入后序遍历
	for (int i = 0; i < n; i++)cin >> ior[i];//输入中序遍历
	for (int i = 0; i < n; i++)m[ior[i]] = i;
	TreeNode* T = build(ior, fin, 0, n - 1, 0, n - 1);//构建二叉树
	travel(T);
}