L2-006 树的遍历 (25 分)（非建树方法）

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

作者: 陈越
单位: 浙江大学
时间限制: 400 ms
内存限制: 64 MB
代码长度限制: 16 KB

输入格式：

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

做题思路：
后序遍历的最后一个元素为树的根节点，可以通过根节点将中序遍历分为左右子树两个部分，在通过中序遍历序列节点的位置，再将后续遍历序列分为左右子树，再递归地求解左右子树的左右子树，并且将每个节点分层存入二维数组中（每递归一次层数+1），分层输出即可。

#include <iostream>
#include <vector>
using namespace std;
vector< vector<int> > ans(30, vector<int>());//建立二维向量存储结果
void Refactor(vector<int>in, vector<int>pos,int cnt)//中序遍历序列，后续遍历序列，当前层序遍历层数
{
	vector<int>inL, inR, posL, posR;//中序遍历左子树，中序遍历右子树，后序遍历左子树，后序遍历右子树
	for (int i = 0; i < in.size(); i++)
	{
		if (pos[pos.size() - 1] == in[i])//当找到树根节点
		{
			ans[cnt].push_back(in[i]);//将当前节点存入第cnt层
			//将当前遍历序列分成左右子树两个部分
			for (int j = 0; j < i; j++)
				inL.push_back(in[j]);
			for (int j = i+1; j < in.size(); j++)
				inR.push_back(in[j]);
			for (int j = 0; j < i; j++)
				posL.push_back(pos[j]);
			for (int j = i; j < pos.size() - 1; j++)
				posR.push_back(pos[j]);
			int temp = cnt + 1;//层数+1
			//先递归重构左子树，再递归重构右子树
			Refactor(inL, posL, temp);
			Refactor(inR, posR, temp);
			return;//重构结束，返回
		}
	}
}
int main()
{
	int n, a, b;
	vector<int>in, pos;
	cin >> n;
	for (int i = 0; i < n; i++)
	{
		cin >> a;
		pos.push_back(a);
	}
	for (int i = 0; i < n; i++)
	{
		cin >> b;
		in.push_back(b);
	}
	Refactor(in, pos, 0);
	int print[30], k = 0;
	for (int i = 0; i < ans.size(); i++)//将数据按层序遍历顺序逐层存入数组ans
	{
		for (int j = 0; j < ans[i].size(); j++)
		{
			print[k] = ans[i][j];
			k++;
		}
	}
	for (int i = 0; i < k; i++)//循环输出结果
	{
		cout << print[i];
		if (i != k - 1)//去除行尾多余空格
			cout << " ";
	}
}