【剑指offer】二叉树的镜像

转载请注明出处：http://blog.csdn.net/ns_code/article/details/25915971

题目描述：

输入一个二叉树，输出其镜像。

输入：

输入可能包含多个测试样例，输入以EOF结束。
对于每个测试案例，输入的第一行为一个整数n(0<=n<=1000,n代表将要输入的二叉树节点的个数（节点从1开始编号）。接下来一行有n个数字,代表第i个二叉树节点的元素的值。接下来有n行，每行有一个字母Ci。
Ci=’d’表示第i个节点有两子孩子，紧接着是左孩子编号和右孩子编号。
Ci=’l’表示第i个节点有一个左孩子，紧接着是左孩子的编号。
Ci=’r’表示第i个节点有一个右孩子，紧接着是右孩子的编号。
Ci=’z’表示第i个节点没有子孩子。

输出：

对应每个测试案例，
按照前序输出其孩子节点的元素值。
若为空输出NULL。

样例输入：

7
8 6 10 5 7 9 11
d 2 3
d 4 5
d 6 7
z
z
z
z

样例输出：

8 10 11 9 6 7 5

    这道题目与上道题目一样，采用数组来作为保存二叉树节点的数据结构，会使测试代码更容易编写。具体的实现算法在代码中比较容易看懂，不细说了，主要想说下，测试时遇到的一些问题（最后AC了）：     1、先序遍历的输出要符合要求的格式，即对于每一个测试用例来说，输出的各节点元素间要有一个空格，且输出最后一个元素后要没有空格，但要输出一个换行符，这就要判断出程序什么时候遍历输出最后一个元素，再递归实现的先序遍历函数中，我没想到能使最后一个元素符合输出格式的方法，因此我这里定义了一个int数组，用来顺序保存先序遍历的节点的data，这样直接遍历输出该int数组即可，并再循环输出中判断要输出的元素是不是最后一个输出元素，如果是，就加上换行符。

    2、由于要输入字符，因此要考虑输入缓冲队列的问题，每行含有字符的数据输入后，会留下换行符，这样会在下次输入字符时被读取，造成读取字符的错误，因此在下一行数据开始输入前，要刷新缓冲区，刚开始图方面，用了fflush(stdin)，在VC++上测试通过了，但是在九度OJ上报了RE，只有一组测试用例通过，多次测试了几组数据之后，怀疑应该是fflush(stdin)的错误，因为有印象在C primer plus上说过，这不是C语言标准里的东西，想gcc应该不支持，后来在网上查了下，gcc是不支持的，这是微软在自家编译器里加的，因此果断用while--getchar()来代替，替换后果然AC了，哎！还是水啊，想想离校招还有不到4个月，心里就没底！

    AC代码：

#include<stdio.h>
#include<stdlib.h>

typedef struct BTNode
{
	int data;
	int rchild;
	int lchild;
}BTNode;


/*
判断pTree1是否包含pTree2
*/
void MirrorTree(BTNode *pTree,int index)
{
	if(pTree == NULL)
		return;
	if(index == -1)
		return;
	if(pTree[index].lchild==-1 && pTree[index].rchild==-1)
		return;

	int temp = pTree[index].lchild;
	pTree[index].lchild = pTree[index].rchild;
	pTree[index].rchild = temp;

	if(pTree[index].lchild!=-1)
		MirrorTree(pTree,pTree[index].lchild);
	if(pTree[index].rchild!=-1)
		MirrorTree(pTree,pTree[index].rchild);
}

//这里设置全局变量i的目的是为了将先序遍历到的元素依次保存到数组preTraverse中，
//其目的是为了确定最后一个输出元素，这样好调整输出的格式，使输出符合测试系统的要求。
int i = 0;
void pre_traverse(BTNode *pTree,int index,int *preTraverse,int n)
{
	if(pTree == NULL)
		return;
	if(index != -1 && i<n)
	{
		preTraverse[i++] = pTree[index].data;
		if(pTree[index].lchild != -1)
			pre_traverse(pTree,pTree[index].lchild,preTraverse,n);
		if(pTree[index].rchild != -1)
			pre_traverse(pTree,pTree[index].rchild,preTraverse,n);
	}
}

int main()
{
	int n;
	while(scanf("%d",&n) != EOF)
	{
		//输入树pTree各节点的值
		BTNode *pTree = NULL;
		if(n>0)
		{
			pTree = (BTNode *)malloc(n*sizeof(BTNode));
			if(pTree == NULL)
				exit(EXIT_FAILURE);
			int i,data;
			//输入n个节点的data
			for(i=0;i<n;i++)
			{
				scanf("%d",&data);
				pTree[i].data = data;
				pTree[i].rchild = -1;
				pTree[i].lchild = -1;
			}

			//输入n行节点连接关系
			for(i=0;i<n;i++)
			{
				char ci;
				//这两行的作用是为了跳过缓冲区中的换行符
				//这里不能用fflush(stdin)，gcc不支持，这只是一些编译器自加的，
				//如果用了，测试系统会报RE。
				while(getchar() != '\n')
					continue;
				scanf("%c",&ci);
				if(ci == 'z')
					continue;
				else if(ci == 'l')
				{
					int lindex;
					scanf("%d",&lindex);
					pTree[i].lchild = lindex-1;
				}
				else if(ci == 'r')
				{
					int rindex;
					scanf("%d",&rindex);
					pTree[i].rchild = rindex-1;
				}
				else if(ci == 'd')
				{
					int lindex,rindex;
					scanf("%d",&lindex);
					scanf("%d",&rindex);
					pTree[i].lchild = lindex-1;
					pTree[i].rchild = rindex-1;
				}
			}
		}
		
		if(n == 0)
		{
			printf("NULL\n");
			continue;
		}
		MirrorTree(pTree,0);
		//先将遍历的元素依次保存到preTraverse数组中
		int *preTraverse = (int *)malloc(n*sizeof(int));
		if(preTraverse == NULL)
			exit(EXIT_FAILURE);
		pre_traverse(pTree,0,preTraverse,n);
		int i;
		for(i=0;i<n;i++)
		{
			//是输出符合测试系统要求的格式
			if(i == n-1)
				printf("%d\n",preTraverse[i]);
			else
				printf("%d ",preTraverse[i]);	
		}

		free(preTraverse);
		preTraverse = NULL;

		free(pTree);
		pTree = NULL;
	}
	return 0;
}

/**************************************************************

     Problem: 1521

     User: mmc_maodun

     Language: C

     Result: Accepted

     Time:0 ms

     Memory:916 kb

****************************************************************/

    写点小插曲：今天AC了这道题后，顺道去科技楼看了下电影《年少轻狂》的拍摄，剧组进驻学校拍摄已经有一周多了，偶尔会在校园的某个地方碰到一些戏份的拍摄，这些天的晚上一直在科技楼前拍南海大学（剧中虚拟的大学）毕业舞会的镜头，虽说实验室就在科技楼，但晚上在图书馆自习，所以一直没去看过，今晚突发奇想，跑去瞄了两眼，听周围人说，前面领舞的女的就是陈妍希，我怎么看都看不出来，楼底下更是停了各种豪车：保时捷跑车、宾利老爷车。。。。尼玛演员、导演就是有钱，我等屌丝敲十几年代码都敲不出一辆宾利来，人家一部戏下来，神马都有了！

   过来瞄几眼，就把我今天的跑步给耽误了，那是别人的舞台，与我无关，回去继续敲代码。。。。。。