二叉树的简单应用

#include<iostream>
#include<stdlib.h>
using namespace std;

typedef struct node{
	int data;
	struct node *left,*right;
}BTNode;

BTNode *CreateBTree(int a[],int n)
{
	BTNode *root,*c,*p,*q;
	root=(BTNode *)malloc(sizeof(BTNode));//单独处理根结点 
	root->data=a[0];
	root->left=root->right=NULL;
	for(int i=1;i<n;i++){//建立链接其他结点 
		p=(BTNode *)malloc(sizeof(BTNode));
		p->data=a[i];
		p->left=p->right=NULL;
		c=root;
		while(c){//找到要插入的位置 
		q=c;
		if(c->data>p->data){//比当前根结点小，从左子树进入 
			c=c->left;
			}
		else{//比当前根结点大，从右子树进入 
			c=c->right;
			}
		}
		if(q->data>p->data){//待插入结点比尾结点的值小，链接在左边 
			q->left=p;
		}
		else{
			q->right=p;
		}
	}
	return root;
}

int LeafCount(BTNode *root)
{
		int L1,L2; 
		if(!root){
			return 0;
		}
		else{
			L1=LeafCount(root->left);//递归遍历左子树 
			L2=LeafCount(root->right);//递归遍历右子树 
			if(!root->left&&!root->right){//判断是否为叶子结点 
				return L1+L2+1;//是的话，加一 
			}
			else{
				return L1+L2;//不是加0 
			}/*return L1+L2+(!root->left&&!root->right)*/
		}
}

int ParentCount(BTNode *root)
{
		int L1,L2; 
		if(!root){
			return 0;
		}
		else{
			L1=ParentCount(root->left);//递归遍历左子树 
			L2=ParentCount(root->right);//递归遍历右子树 
			if(root->left&&root->right){//判断是否为双分支结点 
				return L1+L2+1;//是的话，加一 
			}
			else{
				return L1+L2;//不是加0 
			}
		}
}

int LevelCount(BTNode *root)
{
	int L1,L2;
	if(!root){
		return 0;
	}
	else{
		L1=LevelCount(root->left);
		L2=LevelCount(root->right);
		return L1>L2?L1+1:L2+1;//判断两者谁大，大的一方加一 
	}
}

void MirrorBTree(BTNode *root)
{
	BTNode *t;
	if(root){
		t=root->left;
		root->left=root->right;
		root->right=t;
		MirrorBTree(root->left);
		MirrorBTree(root->right);
	}
} 

void PreOrder(BTNode *root)
{
	if(root){
	cout<<root->data<<" ";
	PreOrder(root->left);
	PreOrder(root->right);
	}
} 
int main()
{
	int n,cnt;
	cout<<"请输入数组长度n：";
	cin>>n;
	int *a;
	a=(int *)malloc(n*sizeof(int));
	cout<<"请输入数组元素:";
	for(int i=0;i<n;i++){
		cin>>a[i];
	}
	BTNode *root;
	//建立二叉排序树 
	root=CreateBTree(a,n);
	cnt=LeafCount(root);//求叶子结点个数 
	cout<<cnt<<endl;
	cnt=ParentCount(root);//求双分支结点个数
	cout<<cnt<<endl; 
	cnt=LevelCount(root);
	cout<<cnt<<endl;
	MirrorBTree(root);//镜像树
	PreOrder(root);//前序遍历验证是否转为镜像树 
	return 0;
}