团体程序设计天梯赛GPLT——L2-006 树的遍历

点击转到PTA题目

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

输入格式：
输入第一行给出一个正整数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<bits/stdc++.h>
using namespace std;
const int N=10005;
int n;
int Pre_Order[N];
int In_Order[N];
int Post_Order[N];
void calculate(int i,int j,int root)//想一想为什么必须要有根这个参数
{
	if(i>j) return;
	int a=i,b=j;
	while(b>i&&Post_Order[root]!=In_Order[b]) b--;//寻根
	cout << In_Order[b] << " ";
	calculate(i,b-1,root+b-1-j);
	calculate(b+1,j,root-1);
}
int main()
{
	cin >> n;
	for(int i=0;i<n;i++) cin >> Post_Order[i];
	for(int i=0;i<n;i++) cin >> In_Order[i];	
	calculate(0,n-1,n-1);
	return 0;
}

知道了上面的求法后，我们需要知道怎么求层序遍历。首先可以建树然后利用队列即BFS的那种方式来实现，第二种就是给他建立一个索引，假设主根的索引为0，那么它的左子树的索引就是1，右子树为2，规律就是左子树=父根 * 2+1,右子树就是父根 * 2+2，最终我们在输出的时候以索引从小到大输出即可，如果索引为0说明没有该子树，就不用输出。

#include<bits/stdc++.h>
using namespace std;
const int N=10005;
int n;
int Pre_Order[N];
int In_Order[N];
int Post_Order[N];
int Level_Order[N]; 
void calculate(int i,int j,int root,int index)
{
	if(i>j) return;
	int a=i,b=j;
	while(b>i&&Post_Order[root]!=In_Order[b]) b--;
	Level_Order[index]=In_Order[b];
	calculate(i,b-1,root+b-1-j,index*2+1);
	calculate(b+1,j,root-1,index*2+2);
}
int main()
{
	cin >> n;
	for(int i=0;i<n;i++) cin >> Post_Order[i];
	for(int i=0;i<n;i++) cin >> In_Order[i];	
	calculate(0,n-1,n-1,0);
	int cnt=0;
	for(int i=0;i<10005;i++)
	{
		if(cnt==n) break;
		if(Level_Order[i])
		{
			if(cnt!=n-1)
		  {
			cout << Level_Order[i] << " ";
		    cnt++;
		  }		
		   else{
			cout << Level_Order[i];
			cnt++;
		  }
		}
	}
	return 0;
}

这种很经典的树的遍历问题是我们一定要学会掌握的，考察的可能性比较大。