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

思路：

以上图为例子，后序遍历最后一个8是整棵树的根结点，在中序遍历中找到结点8，我们发现，中序遍历8的左边是整棵树的左子树，右边是整棵树的右子树。再继续寻找后序遍历的第二个结点（6），我们发现它是结点8的右子树的根结点，同时在中序遍历中6的左边为其左子树，右边为其右子树。。。。。。以此类推。

最重要的是会根据后序遍历和中序遍历创建二叉树，学过数据结构的应该都懂。

建树代码：

node* creatTree(int* reversepost, int* reversein, int n)//利用后序和中序的逆置来创建二叉树相对容易
//reversepost是后续遍历的逆置，reversein是中序遍历的逆置
{
	if (n == 0) return NULL;
	int i = 0;
	while (reversepost[0]!= reversein[i]) i++;//在中序序列中寻找根
	node* root = new node(reversepost[0]);//创建根结点
	root->r = creatTree(reversepost+1, reversein,i);//从逆后序reversepost+1开始对中序的0~i-1右子序列的k个元素递归创建右子树
	root->l= creatTree(reversepost+i+1, reversein+i+1, n-i-1);//从逆中序reversepost+i+1开始对逆中序的i+1~n-1右子序列的n-i-1个元素递归创建左子树
	return root;
}

AC代码：

#include<iostream>
#include<string.h>
#include <iomanip>
#include<vector>
#include<queue>
#include <algorithm>
using namespace std;
struct node
{
	int data;
	node* l = NULL;
	node* r = NULL;
	node(int d) :data(d), l(NULL), r(NULL) {}
};
node* creatTree(int* reversepost, int* reversein, int n)//利用后序和中序的逆置来创建二叉树相对容易
//reversepost是后续遍历的逆置，reversein是中序遍历的逆置
{
	if (n == 0) return NULL;
	int i = 0;
	while (reversepost[0]!= reversein[i]) i++;//在中序序列中寻找根
	node* root = new node(reversepost[0]);//创建根结点
	root->r = creatTree(reversepost+1, reversein,i);//从逆后序reversepost+1开始对中序的0~i-1右子序列的k个元素递归创建右子树
	root->l= creatTree(reversepost+i+1, reversein+i+1, n-i-1);//从逆中序reversepost+i+1开始对逆中序的i+1~n-1右子序列的n-i-1个元素递归创建左子树
	return root;
}
void Levelorder(node* root, vector<node *>& v)//层次遍历
{
	queue<node*> q;
	node* p = root;
	q.push(p);
	int k = 0;
	while (!q.empty())
	{
		v.push_back(q.front());//将队头元素进v
		p = q.front();//记录弹出的元素不然就死循环了
		q.pop();//弹出队头元素
		if (p->l != NULL) q.push(p->l);//进左元素
		if (p->r != NULL) q.push(p->r);//进右元素
	}
}

int main()
{
	int N;
	cin >> N;
	int* post = new int[N];
	int* reversepost = new int[N];
	int* in = new int[N];
	int* reversein = new int[N];
	node* root=NULL;
	vector<int> v;
	vector<node *> v1;
	for (int i = 0; i < N; i++)
	{
		cin >> post[i];
		reversepost[N-1-i]= post[i];//逆置后续遍历
	}
	
	for (int i = 0; i < N; i++)
	{
		cin >> in[i];
		reversein[N - 1 - i] = in[i];//逆置中序遍历
	}
	root = creatTree(reversepost, reversein, N);//创建二叉树
	Levelorder(root, v1);//层次遍历

	for (int i = 0; i < N; i++)//输出层次遍历的结果
	{
		if (i == 0) cout << v1[i]->data;
		else cout <<" "<< v1[i]->data;
	}

}