7-3 玩转二叉树

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

输入格式：

输入第一行给出一个正整数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 <stdlib.h>
#include <queue>
using namespace std;

typedef struct BiTNode{  //二叉树的结构类型定义 
	int data;
	struct BiTNode *lchild,*rchild;
}BiTNode,*BiTree;

BiTree InitTree(int *in,int *pre,int N)
{
	if(N<=0)
		return NULL;
	
	BiTree T;
	int index;
	int root = pre[0];
	for(int i=0;i<N;i++)
	{
		if(in[i]==root)
		{
			index = i;
			T = (BiTree)malloc(sizeof(BiTNode));
			T->data = root;
			break;
		}
	}
	
	//递归建树，注意中序和先序的数组从0开始，而N从1开始 
	T->lchild = InitTree(in,pre+1,index);
	T->rchild = InitTree(in+index+1,pre+index+1,N-index-1);
	
	return T;
} 

void LevelOrderTree(BiTree T)
{
	BiTree q;
	queue<BiTree> Q;
	Q.push(T);
	while(Q.empty()==0) //队列不空
	{
		q = Q.front();
		cout<<q->data<<" ";
		Q.pop();
		
		if(q->rchild!=NULL)  //镜面反转，仅需先右子树，再左子树即可 
			Q.push(q->rchild);
		if(q->lchild!=NULL)
			Q.push(q->lchild);
	} 
}


int main()
{
	BiTree T;
	int in[31],pre[31],N;
	
	cin>>N;
	for(int i=0;i<N;i++)
	{
		cin>>in[i];
	}
	
	for(int i=0;i<N;i++)
	{
		cin>>pre[i];
	}
	
	T=InitTree(in,pre,N);

	LevelOrderTree(T);
	
	return 0;
}

河北工业大学