二叉树

实验内容：由先序和中序建立二叉树，二叉树的三种遍历，求二叉树的高度。统计二叉树中度为1、度为2、度为0的个数。

代码：

#include "stdafx.h"
#include<string.h>

typedef char TElemType;
typedef struct BiTNode{
	TElemType data;
	struct BiTNode *lchild,*rchild;
}BiNode,*BiTree;
//建树
void createBiTreeInPre(BiTree& T,TElemType pre[],int preL,int preR,TElemType in[],int inL,int inR)
{
	if(inL>inR)
	{
		T=NULL;
		return ;
	}
	T=new BiNode;
	T->data=pre[preL];
	int i;
	for(i=inL;T->data!=in[i];i++);

	createBiTreeInPre(T->lchild,pre,preL+1,preL+i-inL,in,inL,i-1);
	createBiTreeInPre(T->rchild,pre,preL+i-inL+1,preR,in,i+1,inR);	
}

void NLR(BiTree T)//先序遍历
{
	if(T==NULL) 
	{
		return ;
	}

	printf("%c",T->data);
	NLR(T->lchild);
	NLR(T->rchild);
}
void LNR(BiTree T)//中序遍历
{
	if(T==NULL)
	{
		return ;
	}
	LNR(T->lchild);
	printf("%c",T->data);
	LNR(T->rchild);
}

void LRN(BiTree T)//后序遍历
{
	if(T==NULL)
	{
		return ;
	}
	LRN(T->lchild);
	LRN(T->rchild);
	printf("%c",T->data);
}

void getLeafNum(BiTree T,int &n)//得到叶子节点的个数
{
	if(T==NULL) return ;

	if(T->rchild==NULL&&T->lchild==NULL)
	{
		n++;
	}

	getLeafNum(T->lchild,n);
	getLeafNum(T->rchild,n);
}
void getDeg1(BiTree T,int &n)//得到度为一节点的个数
{
	if(T==NULL)
	{
		return ;
	}
	if(T->rchild==NULL&&T->lchild!=NULL)
	{
		n++;
	}
	if(T->rchild!=NULL&&T->lchild==NULL)
	{
		n++;
	}

	getDeg1(T->lchild,n);
	getDeg1(T->rchild,n);
}
void getDeg2(BiTree T,int &n)// 得到度为二节点的个数
{
	if(T==NULL)
	{
		return ;
	}
	if(T->lchild!=NULL&&T->rchild!=NULL)
	{
		n++;
	}
	getDeg2(T->lchild,n);
	getDeg2(T->rchild,n);
}

int getTH(BiTree T)// 得到树的高度
{
	if(T==NULL)
	{
		return 0;
	}
	if(T->lchild==NULL&&T->rchild==NULL)
	{
		return 1;
	}
	int H1=getTH(T->lchild);
	int H2=getTH(T->rchild);
	if(H1>H2)
	{
		return H1+1;
	}
	else
	{
		return H2+1;
	}
}

int main()
{
	int n;
	BiTree T;
	TElemType pre[]="532149876";
	TElemType in[]="123456789";
	
	createBiTreeInPre(T,pre,0,strlen(pre)-1,in,0,strlen(in)-1);

	printf("先序遍历：");
	NLR(T);
	printf("\n");

	printf("中序遍历：");
	LNR(T);
	printf("\n");

	printf("后序遍历：");
	LRN(T);
	printf("\n");

	n=0;
	getLeafNum(T,n);
	printf("叶子节点的个数为：%d\n",n);

	n=0;
	getDeg1(T,n);
	printf("度为1的节点的个数为：%d\n",n);

	n=0;
	getDeg2(T,n);
	printf("度为2的节点的个数为：%d\n",n);
	
	printf("树的高度为：%d\n",getTH(T));


	return 0;
}