数据结构-树-判断平衡二叉树

数据结构-树-判断平衡二叉树

题目描述

给定一个二叉树，判断它是否是高度平衡的二叉树。

本题中，一棵高度平衡二叉树定义为：

一个二叉树每个节点的左右两个子树的高度差的绝对值不超过1。

示例

给定二叉树 [3,9,20,null,null,15,7]

    3
   / \
  9  20
    /  \
   15   7

return true

给定二叉树 [1,2,2,3,3,null,null,4,4]

       1
      / \
     2   2
    / \
   3   3
  / \
 4   4

return false

核心代码

Status BalanceBinaryTree(BiTree T, BiTree T1) {

	/*临接情况的递归出口*/
	if (T == NULL&&T1 == NULL)return TRUE; 
	if (T == NULL)
		return BalanceBinaryTree(T1->lchild, T1->rchild);
	if (T1 == NULL)
		return BalanceBinaryTree(T->lchild, T->rchild);

	/*递归函数*/
	if (abs(BiTreeDepth(T) - BiTreeDepth(T1)) > 1)
	{
		return FALSE;
	}
	
	/*递归出口、核心判断*/

	if (BalanceBinaryTree(T->lchild, T->rchild)&&BalanceBinaryTree(T1->lchild, T1->rchild)) {
		return TRUE;
	}
	else {
		return FALSE;
	}

}

可运行源代码

#include "string.h"
#include "stdio.h"    
#include "stdlib.h"   
#include "io.h"  
#include "math.h"  
#include "time.h"

#define OK 1
#define ERROR 0
#define TRUE 1
#define FALSE 0

#define MAXSIZE 100 /* 存储空间初始分配量 */

typedef int Status;		/* Status是函数的类型,其值是函数结果状态代码，如OK等 */

						/* 用于构造二叉树********************************** */
int index = 1;
int index1 = 1;
typedef char String[100]; /*  0号单元存放串的长度 */
String str, str1;

/*分配字符串*/
Status StrAssign(String T, char *chars)
{
	int i;
	/*strlen计算字符串的长度*/
	if (strlen(chars)>MAXSIZE)
		return ERROR;
	else
	{
		T[0] = strlen(chars);
		for (i = 1; i <= T[0]; i++)
			T[i] = *(chars + i - 1);
		return OK;
	}
}
/* ************************************************ */

typedef char TElemType;
TElemType Nil = ' '; /* 字符型以空格符为空 */

					 /*访问结点*/
Status visit(TElemType e)
{
	printf("%c ", e);
	return OK;
}

typedef struct BiTNode  /* 结点结构 */
{
	TElemType data;		/* 结点数据 */
	struct BiTNode *lchild, *rchild; /* 左右孩子指针 */
}BiTNode, *BiTree;


/* 构造空二叉树T */
Status InitBiTree(BiTree *T)
{
	*T = NULL;
	return OK;
}

/* 初始条件: 二叉树T存在。操作结果: 销毁二叉树T */
void DestroyBiTree(BiTree *T)
{
	if (*T)
	{
		if ((*T)->lchild) /* 有左孩子 */
			DestroyBiTree(&(*T)->lchild); /* 销毁左孩子子树 */
		if ((*T)->rchild) /* 有右孩子 */
			DestroyBiTree(&(*T)->rchild); /* 销毁右孩子子树 */
		free(*T); /* 释放根结点 */
		*T = NULL; /* 空指针赋0 */
	}
}

/* 按前序输入二叉树中结点的值（一个字符） */
/* #表示空树，构造二叉链表表示二叉树T。 */
void CreateBiTree(BiTree *T)
{
	TElemType ch;

	/* scanf("%c",&ch); */
	ch = str[index++];

	/*左右子树切换*/
	if (ch == '#')
		*T = NULL;
	else
	{
		/*生成树结点*/
		*T = (BiTree)malloc(sizeof(BiTNode));
		/*if (!*T)
		exit(OVERFLOW);*/
		(*T)->data = ch; /* 给结点赋值 */

						 /*递归创建左右子树*/
		CreateBiTree(&(*T)->lchild); /* 构造左子树 */
		CreateBiTree(&(*T)->rchild); /* 构造右子树 */
	}
}





/* 初始条件: 二叉树T存在。操作结果: 返回T的深度 */
int BiTreeDepth(BiTree T)
{
	int i, j;
	if (!T)
		return 0;
	if (T->lchild)
		/*递归调用*/
		i = BiTreeDepth(T->lchild);
	else
		i = 0;
	if (T->rchild)
		/*递归调用*/
		j = BiTreeDepth(T->rchild);
	else
		j = 0;
	return i>j ? i + 1 : j + 1;
}





Status BalanceBinaryTree(BiTree T, BiTree T1) {

	/*递归函数*/
	if (T == NULL&&T1 == NULL)return TRUE; 
	if (T == NULL)
		return BalanceBinaryTree(T1->lchild, T1->rchild);
	if (T1 == NULL)
		return BalanceBinaryTree(T->lchild, T->rchild);


	if (abs(BiTreeDepth(T) - BiTreeDepth(T1)) > 1)
	{
		return FALSE;
	}
	
	/*递归出口、核心判断*/

	if (BalanceBinaryTree(T->lchild, T->rchild)&&BalanceBinaryTree(T1->lchild, T1->rchild)) {
		return TRUE;
	}
	else {
		return FALSE;
	}

}

int main()
{
	int i;
	BiTree T;
	TElemType e1;


	/*初始化二叉树*/
	InitBiTree(&T);

	/*分配字符串*str=string，目的利用字符串给二叉树赋值*/
	/*前序输入，若不按规则输入，将报异常*/
	/*#代表空结点*/
	//StrAssign(str, "ABC##D##BD##C##");
//	StrAssign(str, "ABCD####E##");
	StrAssign(str, "ABC##D##BDAA###B##C##");

	/*创建二叉树*/
	CreateBiTree(&T);


	int k = BalanceBinaryTree(T->lchild, T->rchild);
	//DispBtree(T); 
	if (k == TRUE) {
		printf("TRUE");
	}
	else {
		printf("FALSE");
	}

	//DispBtree(T);
	/*销毁二叉树*/
	//DestroyBiTree(&T);

	system("pause");
}

运行结果