【面试题】剑指Offer-39-求二叉树的深度和判断一颗树是否为平衡二叉树

题目概述

题目1：求二叉树的深度

题目2：判断一颗二叉树是否为平衡二叉树

解题思路

求二叉树的深度

二叉树的深度呢，这个问题比较简单

递归遍历一遍二叉树，每到一个节点，便返回左子树和右子树深度较大的那个

并+1返回

判断一颗树是否为平衡二叉树

这里呢，也要用到深度

所谓的平衡二叉树，就是左子树的深度和右子树的深度相差的绝对值小于2

每到一个节点，便定义leftDepth和rightDepth

并用先序遍历向左递归，左子树遍历完了后向右递归

如此，每个节点的左右子树的深度都可以记下来

通过判断，便可以知道是否为平衡二叉树了

代码实现

求二叉树的深度

//求二叉树的深度
size_t Depth(BTNode* root)
{
	if (root == NULL)
		return 0;

	int leftDepth = Depth(root->left);
	int rightDepth = Depth(root->right);

	return leftDepth > rightDepth ? leftDepth + 1 : rightDepth + 1;
}

判断一颗二叉树是否为平衡二叉树

//判断一颗二叉树是否为平衡二叉树
bool IsBalance(BTNode* root, size_t& depth)
{
	if (root == NULL)
	{
		depth = 0;
		return true;
	}

	size_t leftDepth = 0;
	size_t rightDepth = 0;

	bool LeftIsBalance = IsBalance(root->left, leftDepth);
	bool RightIsBalance = IsBalance(root->right, rightDepth);

	bool CT_IsBalance = LeftIsBalance && RightIsBalance;

	if (CT_IsBalance)
	{
		depth = leftDepth > rightDepth ? leftDepth + 1 : rightDepth + 1;
		return (leftDepth - rightDepth)< 2;
	}

	return false;
}