给定一个二叉树,判断它是否是高度平衡的二叉树。
一棵高度平衡二叉树定义为:一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过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);
}