SWUST OJ 1077: 平衡二叉树的判定

题目描述

编写程序判断给定的二叉树是否是平衡二叉树。

输入

二叉树的先序序列。

输出

如果是平衡二叉树，输出yes！，否者输出no！

样例输入

AB##C##

样例输出

yes！

参考程序

#include<stdio.h>
#include<malloc.h>
#include<math.h>
typedef struct node
{
	char data;
	int factor;   //当前节点的平衡因子
	struct node *lchild,*rchild;
}BTnode;

BTnode *create()
{
	char ch;
	BTnode *b;
	scanf("%c",&ch);
	if(ch=='#')
		return NULL;
	else
	{
		b=(BTnode *)malloc(sizeof(BTnode));
		b->lchild=create();
		b->rchild=create();
	}
	return b;
}

int flag=1;
int cal(BTnode *b)  //计算每一个节点的平衡因子
{
	if(b==0) return 0;
	int l,r,h;
	l=cal(b->lchild);
	r=cal(b->rchild);
	h=l>r?(l+1):(r+1);
	b->factor=l-r;
	if(b->factor<-1||b->factor>1) flag=0;
	return h;
}

int main()
{
	BTnode *b=NULL;
	b=create();
	cal(b);
	if(flag) printf("yes!");
	else printf("no!");
	return 0;
}

注意

该程序仅供学习参考！