申明
再次申明这个专栏只是分享博主学习图论时的顺序,给新手提供一个入门的好方法(博主现在也是新手所以博主写的代码适合C转C++的新手学习),具体算法逻辑需要自己学习,代码区域有部分解释,因此有错误可以私信或者直接在评论区骂。
前言
上文我们学到了二叉树的前序,中序,后序遍历以及二叉树的存储方式,现在我们将继续学习二叉树已知后序中序排列求层序的方法,我们需要了解后序中序的性质,并进行dfs建好树,然后用bfs来层序遍历树。
思路
因为本题较为经典因此我们深入讲解,后序遍历的顺序是左右根,中序遍历的顺序是左根右,因此后序遍历的最右边就是这个树的根节点,我们通过找到这个根节点在中序遍历中的位置,就能确定左子树的范围,和右子树的范围,我们再dfs递推重新进行这个过程。
题目链接
PAT甲级真题1020 树的遍历
代码
#include<iostream>
#include<algorithm>
#include<queue>
#include<cstring>
#include<string>
using namespace std;
const int N=35;
int post[N],in[N];
struct tree
{
int left,right;
}q[N];
int n;
int dfs(int il,int ir,int pl,int pr)
{
int root=post[pr];
int k;
for(int i=il;i<=ir;i++)
{
if(in[i]==root) k=i;
}
if(il<k) q[root].left=dfs(il,k-1,pl,pl+k-1-il); //存在左子树
if(ir>k) q[root].right=dfs(k+1,ir,pl+k-1-il+1,pr-1);
return root;
}
int bfs()
{
int x=post[n];
queue<int> qq;
qq.push(x);
while(qq.size())
{
int t=qq.front();
qq.pop();
printf("%d ",t);
if(q[t].left) qq.push(q[t].left);
if(q[t].right) qq.push(q[t].right);
}
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%d",&post[i]);
}
for(int i=1;i<=n;i++)
{
scanf("%d",&in[i]);
}
dfs(1,n,1,n);
bfs();
}