题目来源:Acwing 1497.树的遍历||PTA L2-006 树的遍历 (25 分)
21.04.19补充:回头看这个方法太麻烦了,简单些的可以参考 PTA天梯练习赛 刷题笔记1 的 L2-011 玩转二叉树
题目描述
一个二叉树,树中每个节点的权值互不相同。
现在给出它的后序遍历和中序遍历,请你输出它的层序遍历。输入描述
第一行包含整数 N,表示二叉树的节点数。
第二行包含 N 个整数,表示二叉树的后序遍历。
第三行包含 N 个整数,表示二叉树的中序遍历。输出描述
输出一行 N 个整数,表示二叉树的层序遍历。
数据范围
1≤N≤30
输入样例1
7
2 3 1 5 7 6 4
1 2 3 4 5 6 7输出样例1
4 1 6 3 5 7 2
思路
对于任何一个二叉树的合法中序遍历与后序遍历,中序遍历中 在后序遍历中位置最靠后的 数,即为该树的根;
依照此根,我们又可以将该中序遍历分为左右两部分,即为该根的左右两棵子树,进行递归处理。
建树完成之后,通过BFS获取层序遍历。
代码
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
vector<int> son[31];//son[x]代表x的子节点
int n,i,j;
int mid[31],lst[31],rot=0;//rot存总根
void maketr(int l,int r,int fa)//fa存该树的根的父节点
{
if(l==r)//叶子
{
son[fa].push_back(mid[l]);
return;
}
else if(l>r)return;//非法
int root=0,rm=0;//root存该子树的根,rm存该根在中序遍历中的位置
for(i=l;i<=r;i++)//对于该子树的每一个节点
{
for(j=1;j<=n;j++)//在后序遍历中位置最靠后的一个
{
if(mid[i]==lst[j])
{
if(j>root){root=j,rm=i;}
//printf("*");
break;
}
}
}
if(!rot)rot=lst[root];//如果总根未确定,该根即为总根
//printf("%d\n",root);
son[fa].push_back(lst[root]);//将此树的根连接至其父节点
maketr(l,rm-1,lst[root]);//递归左子树
maketr(rm+1,r,lst[root]);//递归右子树
return;
}
queue<int> que;//bfs队列
void bfs(int x)
{
//printf("*");
int pf=0;
for(i=0;i<son[x].size();i++)que.push(son[x][i]);
while(!que.empty())
{
int now=que.front();
que.pop();
if(!pf)pf=1;
else printf(" ");
printf("%d",now);
for(i=0;i<son[now].size();i++)que.push(son[now][i]);//树无自环、重边与回路,不需要考虑重复加入的问题
}
}
int main()
{
cin>>n;
for(i=1;i<=n;i++)scanf("%d",&lst[i]);
for(i=1;i<=n;i++)scanf("%d",&mid[i]);
maketr(1,n,0);//建树
bfs(0);//bfs输出
return 0;
}
注1:代码未解决行末空格的问题(已解决);
注2:时间复杂度O(n3),待优化。
ED
\