题目描述
假设一个二叉树上各结点的权值互不相同。
我们就可以通过其后序遍历和中序遍历来确定唯一二叉树。
请你输出该二叉树的 Z 字形遍历序列----也就是说,从根结点开始,逐层遍历,第一层从右到左遍历,第二层从左到右遍历,第三层从右到左遍历,以此类推。
输入格式
第一行包含整数 N,表示二叉树结点数量。
第二行包含 N 个整数,表示二叉树的中序遍历序列。
第三行包含 N 个整数,表示二叉树的后序遍历序列。
输出格式
输出二叉树的 Z 字形遍历序列。
样例
样例输入
8
12 11 20 17 1 15 8 5
12 20 17 11 15 8 5 1
样例输出
1 11 5 8 17 12 20 15
数据范围与提示
1≤N≤300
1.分析题目
首先这题考的是二叉树的遍历(谁都知道)
先用中序和后序求出整棵树每个节点的左右儿子,
然后,BFS。再然后,我就Wa了,又仔细看了看,要Z字型的遍历,就想到了双端队列(duque)。
2.duque
双端队列(duque)是c++STL中的一个容器,与普通队列不同,双端队列可以在两头快速插入与删除。
3.代码实现
#include<bits/stdc++.h>
using namespace std;
struct Terr{
int data,L,R;
}s[305];//定义Terr结构体来存放树
int n;
int b[305];
int dfs(int la,int ra,int lb,int rb){
if(la>ra)return 0;
for(int i=la;i<=ra;i++){
if(s[i].data==b[rb]){
s[i].L=dfs(la,i-1,lb,lb+i-la-1);
s[i].R=dfs(i+1,ra,lb+i-la,rb-1);//递归
return i;
}
}
}///通过DFS递归来用中序和后序表示树
void bfs(int x){
int q[1005];
int l,r;
r=l=500;//数组模拟双端队列
q[++r]=x;//根节点入队列
bool f=1;//标记遍历方向
while(l<r){
if(f){
int y=r-l;//获得队列长度
for(int i=1;i<=y;i++){
cout<<s[q[l+1]].data<<' ';//输出
if(s[q[l+1]].R!=0)q[++r]=s[q[l+1]].R;
if(s[q[l+1]].L!=0)q[++r]=s[q[l+1]].L;//队头元素左右子节点入队列
l++;//队头元素出队列
}
f=0;//方向取反
}
else{
int y=r-l;//获得队列长度
for(int i=1;i<=y;i++){
cout<<s[q[r]].data<<' ';
if(s[q[r]].L!=0)q[l]=s[q[r]].L,l--;
if(s[q[r]].R!=0)q[l]=s[q[r]].R,l--;//对头元素左右子节点入队列
r--;//队头元素出队列
}
f=1;//方向取反
}
}
}
int main()
{
cin>>n;//树的结点数
for(int i=1;i<=n;i++)cin>>s[i].data;//节点权值
for(int i=1;i<=n;i++)cin>>b[i];//后序遍历
dfs(1,n,1,n);//DFS确定树
for(int i=1;i<=n;i++){
if(s[i].data==b[n]){//寻找根节点
bfs(i);//BFS输出
return 0;//结束
}
}
}