给定一棵二叉树的后序遍历和中序遍历,请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。
输入格式:
输入第一行给出一个正整数N(≤30),是二叉树中结点的个数。第二行给出其后序遍历序列。第三行给出其中序遍历序列。数字间以空格分隔。
输出格式:
在一行中输出该树的层序遍历的序列。数字间以1个空格分隔,行首尾不得有多余空格。
输入样例:
7 2 3 1 5 7 6 4 1 2 3 4 5 6 7
输出样例:
4 1 6 3 5 7 2
思路:因为后序遍历最后的一定是根,就在中序里找并以其为中点,分离出左右两个子树,再在子树里再次这样操作。p是对结点进行编号的,放在map类型里,对应其在结果数组 c 里的位置。因为map会内部排序,所以最后按map类型里的顺序把结果输出。
注意:变量一定要写对,我就是调用 fund 函数时把 ar 写成了 al, 之后近半个多小时都在找bug。难过~
#include<iostream>
#include<map>
using namespace std;
int a[30],b[30],c[30]={0},cnt=0;
map<int,int> mp;
void fund(int al,int ar,int bl,int br,int p){
int i=bl;
if(al>ar||bl>br)return;
while(b[i]!=a[ar])i++;
mp[p]=cnt;
c[cnt++]=b[i];
fund(al,al+(i-bl)-1,bl,i-1,2*p+1);
fund(ar-(br-i),ar-1,i+1,br,2*p+2);
}
int main(){
int n;
cin>>n;
for(int i=0;i<n;i++)cin>>a[i];
for(int i=0;i<n;i++)cin>>b[i];
fund(0,n-1,0,n-1,0);
for(map<int,int>::iterator it=mp.begin();it!=mp.end();it++){
if(it!=mp.begin())cout<<' ';
cout<<c[it->second];
}
return 0;
}