L2-011. 玩转二叉树

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

输入格式：

输入第一行给出一个正整数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<cstdio>
#include<iostream>
#include<vector> 
#include<queue>
using namespace std;
int pre[40],in[30];
struct node{
	int num;
	struct node* left;
	struct node* right;
}; 
struct node*build(int p1,int p2,int i1,int i2){
	if(p1>p2||i1>i2)return NULL;
	int p;
	for(int i=i1;i<=i2;i++){
		if(in[i]==pre[p1]){
			p=i;
			break;
		}
	}
	struct node* root=new node();
	root->num=in[p];
	root->left=build(p1+1,p1+p-i1,i1,p-1);
	root->right=build(p1+p-i1+1,p2,p+1,i2);
	return root; 
}
queue<struct node*>q;
vector<int>vec;
void bfs(struct node * root){
	q.push(root);
	while(!q.empty()){
		node* tmp=q.front();
		q.pop();
		vec.push_back(tmp->num);
		if(tmp->right)q.push(tmp->right);
		if(tmp->left)q.push(tmp->left);
	}
}
int main(){
	int n;
	scanf("%d",&n);
	for(int i=0;i<n;i++){
		scanf("%d",&in[i]);
	}
	for(int i=0;i<n;i++){
		scanf("%d",&pre[i]);
	}
	struct node*root=NULL;
	root=build(0,n-1,0,n-1);
	bfs(root);
	for(int i=0;i<vec.size()-1;i++){
		printf("%d ",vec[i]);
	}
	printf("%d\n",vec[vec.size()-1]);
	return 0;
}