平衡二叉树的判定
题目内容:编写程序判断给定的二叉树是否是平衡二叉树。
输入
二叉树的先序序列。
输出
如果是平衡二叉树,输出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;
}