PTA 玩转二叉树

给定一棵二叉树的中序遍历和前序遍历，请你先将树做个镜面反转，再输出反转后的层序遍历的序列。所谓镜面反转，是指将所有非叶结点的左右孩子对换。这里假设键值都是互不相等的正整数。

输入格式：

输入第一行给出一个正整数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

 根据前序和中序递归建树

#include<bits/stdc++.h>
using namespace std;
int tree[100086];
int xx[10086];
int zx[10086];
int ans[10086];
map<int,int>mp;
int n;
void build(int x,int a,int b,int c,int d){
	if(b<a || d  < c){
		return ;
	}
	if(a==b){
		tree[x]=zx[a];
		return ;
	}
	for(int i=a;i<=b;i++){
		if(zx[i]==xx[c]){
			tree[x]=xx[c];
			build(2 * x , a , i-1 , c+1 , c+i-a );
			build(2*x+1 , i+1 , b , c+i-a+1 , d );
		}
	}
	return ;
	
}
void bfs(int x){
	queue<int>q;
	int cnt =1;
	q.push(x);
	while(!q.empty()){
		int now = q.front();
		q.pop();
		if(tree[now]!=0){
			ans[cnt]=tree[now];
			cnt++;
			q.push(2*now+1);
			q.push(2*now);
		}

	}
}
int main(){
	cin>>n;
	memset(tree,0,sizeof tree);
	for(int i=1;i<=n;i++){
		cin>>zx[i];
	}
	for(int i=1;i<=n;i++){
		cin>>xx[i];
		mp[xx[i]]=i;
	}
	build(1,1,n,1,n);
	bfs(1);
	for(int i=1;i<=n;i++){
		if(i!=1)cout<<" ";
		cout<<ans[i];
	}
	cout<<endl;
	
	
	return 0;
}