平衡二叉树ON解法

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

一棵高度平衡二叉树定义为:一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1

O(n^2)解法

int maxDepth(struct TreeNode* root)//二叉树的最大深度
{
	if (root == NULL)
	{
		return 0;
	}
	int left = maxDepth(root->left) + 1;
	int right = maxDepth(root->right) + 1;

	return left > right ? left : right;
}
bool isBalanced(struct TreeNode* root)//判断是不是平衡二叉树
{
	if (root == NULL) return true;

	int maxleft = maxDepth(root->right);
	int maxright = maxDepth(root->left);

	return abs(maxleft - maxright)<2 && isBalanced(root->right) && isBalanced(root->left);
}

O(n)解法

bool _isBalanced(struct TreeNode* root, int* pDepth) {
	if (root == NULL) {
		*pDepth = 0;
		return true;
	}

	int leftDepth = 0, rightDepth = 0;
	if (_isBalanced(root->left, &leftDepth)
		&& _isBalanced(root->right, &rightDepth)
		&& abs(leftDepth - rightDepth) < 2)
	{
		*pDepth = leftDepth > rightDepth ? leftDepth + 1 : rightDepth + 1;
		return true;
	}
	else
	{
		return false;
	}

}

bool isBalanced(struct TreeNode* root) {
	int depth = 0;
	return _isBalanced(root, &depth);
}

猜你喜欢

转载自blog.csdn.net/lucky52529/article/details/85078562