题目链接:PTA | 程序设计类实验辅助教学平台
分析:这道题分为三个步骤:
(1)根据中序遍历序列和前序遍历序列进行建树
(2)将树进行反转
(3)对树进行层序遍历
先来说一下如何根据中序遍历序列和前序遍历序列进行建树,由前序遍历的方式我们可以知道,先遍历根然后依次遍历左孩子和右孩子,由于建树我们肯定是先建立根再依次建立孩子节点,所以我们直接按照给定的前序遍历进行一个一个插入树中即可(一开始为空树),而根节点刚好把中序遍历分为三部分,根左边的为左子树,根,根右边的为右子树,这样我们就可以进行递归建树了,当某一部分长度为1时就可以直接进行赋值。
下面来说一下反转,这个比较简单,就是直接按照后序遍历的方式,然后直接将左右孩子交换即可
最后是层序遍历,这个是用队列实现的,详情见代码:
#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
#include<map>
#include<cmath>
#include<queue>
using namespace std;
const int N=1e5+10;
int a[N],b[N],t[N];
int l[N],r[N],idx;
bool vis[N];
void build(int pos,int L,int R,int val)
{
if(!vis[pos]||L==R)
{
t[pos]=val;
vis[pos]=true;
return ;
}
int i,j;
for(i=L;i<=R;i++)//找待插入值在中序遍历中的位置
if(a[i]==val) break;
for(j=L;j<=R;j++)//找当前值在中序遍历中的位置
if(a[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 dfs(int pos)//反转
{
if(l[pos]) dfs(l[pos]);
if(r[pos]) dfs(r[pos]);
swap(l[pos],r[pos]);
}
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;
cin>>n;
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
idx=1;
for(int i=1;i<=n;i++)
{
scanf("%d",&b[i]);
build(1,1,n,b[i]);
}
dfs(1);
bfs(1);
return 0;
}