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 n,a[40],b[40];
struct node{
	node *l;
	node *r;
	int val;
};
//先序为根，中序为左右树
//len用来确定右子树的长度
node* tree(int a[],int b[],int len){
	if(len<=0)return NULL; 
	int cnt=0;
	for(int i=0;i<len;i++){
		if(b[0]==a[i]){
			cnt=i;
			break;
		}
	}
	node* root=new node();
	root->val=a[cnt];
	root->l=tree(a,b+1,cnt);
	root->r=tree(a+cnt+1,b+cnt+1,len-cnt-1);
	return root;
}
void ceng(node* root){
	queue<node*> q;
	q.push(root);
	int f=0;
	while(!q.empty()){
		node* tmp=q.front();
		q.pop();
		if(tmp==NULL)continue;
		if(f==0){	
			f=1;
			cout<<tmp->val;
		}else{
			cout<<' '<<tmp->val;
		}
		//题目要求翻转，相当于是反着跑层序遍历，先右再左 
		q.push(tmp->r);
		q.push(tmp->l);
	}
}
int main(){
	cin>>n;
	for(int i=0;i<n;i++) cin>>a[i];
	for(int i=0;i<n;i++) cin>>b[i];
	node* root=tree(a,b,n);
	ceng(root);
	return 0;
}