天梯赛L2-006 树的遍历/PAT甲级 1020 Tree Traverals

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
结尾无空行

初学树，感觉这道题很经典，能归为一大类题
后序遍历已知最后一个节点是根节点，我们又得还原数求层序遍历，利用中序遍历能把二叉树分为左右子树和根节点，我们只需要在中序遍历中找到与根节点相同的节点，再递归构造就好

#include <iostream>
#include <queue>
using namespace std;
struct node {
	int data;
	node* lchild;
	node* rchild;
};
int post[35], in[35];
int n;
//构造树的函数
node* creat(int postL, int postR, int inL, int inR)
{
	//递归边界，后序遍历的左边大于右边，回溯
	if (postL > postR)
		return NULL;
	//存根节点的地址
	node *root = new node;
	root->data = post[postR];
	//找到中序遍历中，根节点的位置
	int k;
	for (k = inL; k < inR; k++)
		if (in[k] == post[postR])
			break;
	//左子树一共有多少个节点
	int numberLeft = k - inL;
	//根据上图规律构造二叉树，后序遍历的左子树序列对应中序遍历的左子树序列
	root->lchild = creat(postL, postL + numberLeft - 1, inL, k - 1);
	//后序遍历右子树序列对应右子树序列
	root->rchild = creat(postL + numberLeft, postR - 1, k + 1, inR);
	return root;
}
int num = 0;
//层序遍历，写法比较死，新建队列把根节点放进去，开循环，把队首元素赋给temp
//pop队首元素，访问temp，根据题意输出
//最后再把下一层的节点放进去，前面的判断别忘了
void BFS(node* root)
{
	queue<node*> qe;
	qe.push(root);
	while (!qe.empty())
	{
		node* temp = qe.front();
		qe.pop();
		cout << temp->data;
		num++;
		if (num < n)
			cout << " ";
		if (temp->lchild != NULL)
			qe.push(temp->lchild);
		if (temp->rchild != NULL)
			qe.push(temp->rchild);
	}
}
int main()
{
	cin >> n;
	for (int i = 0; i < n; i++)
		cin >> post[i];
	for (int i = 0; i < n; i++)
		cin >> in[i];
	node* root = creat(0, n - 1, 0, n - 1);
	BFS(root);
	return 0;
}