给定一棵二叉树的中序遍历和前序遍历,请你先将树做个镜面反转,再输出反转后的层序遍历的序列。所谓镜面反转,是指将所有非叶结点的左右孩子对换。这里假设键值都是互不相等的正整数。
输入格式:
输入第一行给出一个正整数N
(≤30),是二叉树中结点的个数。第二行给出其中序遍历序列。第三行给出其前序遍历序列。数字间以空格分隔。
输出格式:
在一行中输出该树反转后的层序遍历的序列。数字间以1个空格分隔,行首尾不得有多余空格。
输入样例:
7
1 2 3 4 5 6 7
4 1 3 2 6 5 7
输出样例:
4 6 1 7 5 3 2
发现存层序遍历的时候map存会简单很多,而且也能够直接解决镜像问题。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long llu;
const ll N=1e8+7;
int a[37],n;
int b[37];
int check[37];
map<int,int> MoM;
void change(int ml,int mr,int pl,int pr,int h)//中序和先序的左右起始位置
{
int mid;
if(ml>mr||pl>pr||check[pl]) return;
check[pl]=true;
MoM[h]=b[pl];
for(int i=ml;i<=mr;i++)//中序的左子树开始
{
if(a[i]==b[pl]) mid=i;//mid-1为左子树的大小,mid为中序中根的位置
}
change(ml,mid-1,pl+1,pl+1+mid-ml-1,2*h+1);//左开始,放在2*h+1的位置实现镜像
change(mid+1,mr,pl+1+mid-ml,pr,2*h);//右开始
}
int main()
{
cin>>n;
int mid;
for(int i=1;i<=n;i++) cin>>a[i];//zhong
for(int i=1;i<=n;i++){
cin>>b[i];//qian
}
change(1,n,1,n,1);
bool f=false;
for(auto it:MoM)
{
if(f) cout<<" ";
cout<<it.second;
f=true;
}
return 0;
}