题目:给定一个二叉树,判断它是否是高度平衡的二叉树。
方法一:从上到下进行递归
- 先自定义一个函数,用于求二叉树的高度,分别计算二叉树的左右子树的高度
- 判断二叉树是否平衡:左右子树的高度差绝对值小于等于1
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
//从上到下进行递归
/*
1.先自定义一个函数,用于求二叉树的高度,分别计算二叉树的左右子树的高度
2.判断二叉树是否平衡:左右子树的高度差绝对值小于等于1
时间复杂度:O(N*N)
空间复杂度:O(N)
*/
//定义一个求二叉树高度的函数
//如果树为空,则返回0
int height(struct TreeNode* root){
if(root == NULL){
return 0;
}else{
//数不为空,递归算出左右子树的高度再+1(加1是因为根节点占了一个高度)
return fmax(height(root->left),height(root->right)) + 1;
}
}
//判断二叉树是否平衡
bool isBalanced(struct TreeNode* root){
if(root == NULL){
return true;
}else{
//返回的是左右子树的高度差的绝对值是否为1 ,并且左右子树是否分别是平衡树
return fabs(height(root->left) - height(root->right)) <= 1 && isBalanced(root->left) && isBalanced(root->right);
}
}
方法二:从下到上递归
算法思想:从下向上递归,类似于后序遍历,先递归的判断左右子树是否平衡,然后再判断该节点的子树是否平衡,如果一棵子树是平衡的,则返回其高度(高度一定是非负整数),否则返回-1。如果一棵子树不平衡,则整个二叉树一定不平衡
代码如下:
int height(struct TreeNode* root){
if(root == NULL){
return 0;
}
int leftHeight = height(root->left);
int rightHeight = height(root->right);
//如果一棵子树的左右子树不平衡或者整棵树不平衡,返回-1
if(leftHeight == -1 || rightHeight == -1 ||fabs(leftHeight - rightHeight >1 )){
return -1;
}else{
return fmax(leftHeight, rightHeight)+1;
}
}
//判断一棵树是否为平衡二叉树
bool isBalanced(struct TreeNode* root) {
return height(root) >= 0;
}