题目链接:PTA | 程序设计类实验辅助教学平台
之前有一道给出中序遍历和前序遍历让求反转后的树的后序遍历,与这道题目差不多,但是比这道题目多了一个反转操作,详情见(L2-011)玩转二叉树(利用个给定序列建树)_AC__dream的博客-CSDN博客
分析:我们知道后序遍历的方式是先遍历左右孩子最后遍历根,那么也就是说后序遍历序列的结尾是根,通过根可以把中序遍历分成三部分,一部分是根左边的,一个是根,另一部分是根右边的,然后我们再取后序遍历序列中倒数第二个,这又是右子树的一个根,按照这种方法,我们就是倒着插入后续遍历序列中的节点,可以保证每次插入的都是根节点,刚好符合我们建树的思想,最后直接进行层序遍历即可。
下面是代码:
#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
#include<map>
#include<cmath>
#include<queue>
using namespace std;
const int N=1e4+10;
int l[N],r[N],idx,t[N];
int a[N],b[N];
bool vis[N];
void build(int pos,int L,int R,int val)
{
if(!vis[pos]||L==R)
{
vis[pos]=true;
t[pos]=val;
return ;
}
int i,j;
for(i=L;i<=R;i++)//找待插入值在中序遍历中的位置
if(b[i]==val) break;
for(j=L;j<=R;j++)//找当前值在中序遍历中的位置
if(b[j]==t[pos]) break;
if(i<j)//说明待插入节点在以当前节点为根的左子树中
{
if(l[pos])
build(l[pos],L,j-1,val);
else
l[pos]=++idx,build(l[pos],L,j-1,val);
}
else//说明待插入节点在以当前节点为根的右子树中
{
if(r[pos])
build(r[pos],j+1,R,val);
else
r[pos]=++idx,build(r[pos],j+1,R,val);
}
}
void bfs(int x)//求层序遍历
{
queue<int>q;
q.push(x);
printf("%d",t[x]);
while(!q.empty())
{
int tt=q.front();
if(tt!=x)
printf(" %d",t[tt]);
q.pop();
if(l[tt]) q.push(l[tt]);
if(r[tt]) q.push(r[tt]);
}
}
int main()
{
int n;
idx=1;
cin>>n;
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
for(int i=1;i<=n;i++)
scanf("%d",&b[i]);
for(int i=n;i>=1;i--)
build(1,1,n,a[i]);
bfs(1);
return 0;
}