平衡二叉树的判定

平衡二叉树的判定

题目内容：编写程序判断给定的二叉树是否是平衡二叉树。
输入
二叉树的先序序列。
输出
如果是平衡二叉树，输出yes!,否则输出no!
样例输入
AB##C##
样例输出
yes!
下面来看代码：

#include <iostream>
#include <malloc.h>
using namespace std;
struct Tree{
	char data;
	int high;
	Tree *l,*r;
};
void init(Tree *&L)//先序遍历创建二叉树 
{
	char ch;
	cin>>ch;
	if(ch=='#')
	{
		L=NULL;
	}
	else
	{
		L=(Tree*)malloc(sizeof(Tree));
		L->data=ch;
		init(L->l);
		init(L->r);
	}
}
void high(Tree *&L)//求树的高度 
{
	if(L!=NULL)
	{
		high(L->l);//后序遍历 
		high(L->r);
		if(L->l==NULL&&L->r==NULL)
		L->high=1;
		else if(L->l!=NULL&&L->r==NULL)
		{
			L->high=L->l->high+1;
		}
		else if(L->l==NULL&&L->r!=NULL)
		{
			L->high=L->r->high+1;
		}
		else if(L->l!=NULL&&L->r!=NULL)
		{
			if(L->l->high>L->r->high)
			{
				L->high=L->l->high+1;
			}
			else
			{
				L->high=L->r->high+1;
			}
		}
	}
}
int t=1;//立一个flag 
void judge(Tree *&L)
{
	if(L!=NULL)
	{
		if(L->l!=NULL&&L->r!=NULL)
		{
			if(L->l->high-L->r->high!=0&&L->l->high-L->r->high!=1&&L->l->high-L->r->high!=-1)
			{
				t=0;
			}
			else if(L->l!=NULL&&L->r==NULL)
			{
				if(L->l->high!=1)
				t=0;
			}
			else if(L->l==NULL&&L->r!=NULL)
			{
				if(L->r->high!=1)
				t=0;
			}
			judge(L->l);
			judge(L->r);
		}
	}
}
int main()
{
	Tree *L;
	init(L);
	high(L);
	judge(L);
	if(t==1)
	{
		cout<<"yes!";
	}
	else
	{
		cout<<"no!";
	}
	return 0;
}