判断二叉树是否平衡

最新推荐文章于 2021-12-29 21:41:16 发布

moses1213

最新推荐文章于 2021-12-29 21:41:16 发布

阅读量390

点赞数

分类专栏：树与图面试题汇总文章标签：二叉树平衡二叉树

本文链接：https://blog.csdn.net/moses1213/article/details/52020781

版权

树与图面试题汇总专栏收录该内容

8 篇文章 1 订阅

订阅专栏

题目：实现一个函数，检查二叉树是否平衡。在这个问题中，平衡树的定义如下：任意一个结点，其两个子树的高度差不超过1。

解法一：因为要检查每个结点的高度，如果用循环遍历每个结点非常麻烦，所以想到用递归。先判断根结点两个子树的高度差，如果满足再分别判断根结点的两个子结点是否满足平衡。求结点的高度也可以用递归的方法。

int GetHeight(BinaryTreeNode* pRoot)
{
	if(pRoot == NULL)
		return 0;
		
	int leftHeight = GetHeight(pRoot->m_pLeft);
	int rightHeight = GetHeight(pRoot->m_pRight);
	return leftHeight > rightHeight ? leftHeight+1 : rightHeight+1;
}

bool IsBalanceTree(BinaryTreeNode* pRoot)
{
	if(pRoot == NULL)
		return true;
		
	if(abs(GetHeight(pRoot->m_pLeft)-GetHeight(pRoot->m_pRight)) > 1)
		return false;
	else
		return IsBalanceTree(pRoot->m_pLeft) && IsBalanceTree(pRoot->m_pRight);
}

解法二：解法一种求高度时有很多重复计算，譬如求根结点的高度和求它的子结点的高度时重复递归了一段，提高效率的办法是从树的底层往上判断是否满足平衡。

int CheckHeight(BinaryTreeNode* pNode)
{
	if(pNode == NULL)
		return 0;
		
	int leftHeight = CheckHeight(pNode->m_pLeft);
	if(leftHeight == -1)
		return -1;
		
	int rightHeight = CheckHeight(pNode->m_pRight);
	if(rightHeight == -1)
		return -1;
		
	int heightDiff = leftHeight - rightHeight;
	if(abs(heightDiff) > 1)
		return -1;
	else
		return leftHeight > rightHeight ? leftHeight+1 : rightHeight+1;
}

bool IsBalanceTree(BinaryTreeNode* pRoot)
{
	if(CheckHeight(pRoot) == -1)
		return false;
	else
		return true;
}