【天梯赛】L2-011 玩转二叉树 (25分)

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

输入格式：
输入第一行给出一个正整数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 <iostream>
#include <queue>
using namespace std;
int pre[32], in[32];
typedef struct bitnode{
	int data;
	struct bitnode *lchild, *rchild;
}BiTNode, *BiTree;

BiTree buildTree(int root, int start, int end){
	if(start > end) return NULL;
	BiTree p = (BiTNode *)malloc(sizeof(BiTNode));
	p->data = pre[root];
	int i, cnt = 0;
	for(i = start; i < end; i++){
		cnt++;
		if(in[i] == pre[root])
			break;
	}
	p->lchild = buildTree(root+1, start, i-1);
	p->rchild = buildTree(root+cnt, i+1, end);
	return p;
}

queue<BiTree> q;
void Levelorder(BiTree bt){
	q.push(bt);
	int num = 0;
	while(!q.empty()){
		BiTree tmp = q.front();
		if(num > 0) cout << " ";
		cout << tmp->data;
		if(tmp->rchild) q.push(tmp->rchild);
		if(tmp->lchild) q.push(tmp->lchild);
		q.pop();
		num++;
	}
}
int main(){
	int n;
	cin >> n;
	for(int i = 0; i < n; i++){
		cin >> in[i];
	}
	for(int i = 0; i < n; i++){
		cin >> pre[i];
	}
	BiTree p = buildTree(0, 0, n-1);
	Levelorder(p);
}