ptaL2-011 玩转二叉树

ptaL2-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

简要分析：
先序遍历：根 左 右
中序遍历：左 根 右
后序遍历：左 右 根

先根据先序遍历确定根，再根据中序遍历确定左子树和右子树， 最后递归，建成一棵树。根据满二叉树的性质，若i为根节点的序号，则左孩子的序号为2i+1,右孩子的序号为2i+2得该二叉树的层次遍历。题意要求镜面反转，只需将左孩子和右孩子的序号反转即左孩子的序号为2i+2,右孩子的序号为2i+1，便可得到题意需要的层次遍历。

答案如下：

#include <iostream>
#include <cstring>
using namespace std;

/* run this program using the console pauser or add your own getch, system("pause") or input loop */
int n, a[10000], b[10000], c[10000];

void fun(int beginA, int endA, int beginB, int endB, int k){
	if(beginA>endA||beginB>endB)
	return;
	int i = beginA;
	while(a[i]!=b[beginB]){
		i++;
	}
	c[k] = a[i];
	fun(beginA,i-1,beginB+1,beginB-beginA+i,2*k+2);
	fun(i+1,endA,beginB-beginA+i+1,endB,2*k+1);
}

int main(int argc, char** argv) {
	cin>>n;
	for(int i=0;i<n;i++){
		cin>>a[i];
	}
	for(int i=0;i<n;i++){
		cin>>b[i];
	}
	memset(c,-1,sizeof(c));
	fun(0,n-1,0,n-1,0);
	int t = 0;
	for(int i=0;i<=10000;i++){
		if(t>=n)
		break;
		if(c[i]!=-1){
			if(t==0)
			cout<<c[i];
			else
			cout<<" "<<c[i];
			t++;
		}
	}
	return 0;
}