给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。
示例 1:
输入:root = [3,9,20,null,null,15,7]
输出:true
示例 2:
输入:root = [1,2,2,3,3,null,null,4,4]
输出:false
示例 3:
输入:root = []
输出:true
思路:判断一颗二叉树是否是平衡二叉树,要判断二叉树的每一个节点的左右子树的高度差是否大于1。
- 判断一个节点的左右子树的高度差是否大于1,需要分别计算左子树的高度和右子树的高度。计算高度可以用递归实现,与计算一颗树的深度思路相同。
- 自顶向下遍历这颗二叉树,如果前面的节点左右子树的高度差都符合要求,那就继续往下找,看看是不是所有节点的左右子树高度差都符合要求。这里也可以用递归实现:给一个节点,如果这个节点为空,说明左右子树高度都为0,返回true;不为空,计算左右子树高度差。若高度差在要求范围内,判断当前节点的左右子树是否是平衡子树,返回两者&&的结果,即返回当前节点是否是平衡子树;若高度差不符合要求,说明当前子树不是平衡子树,返回false。
代码:
class Solution { //110. 平衡二叉树
public:
bool isBalanced(TreeNode* root) {
if (root == nullptr) return true;
int difference = computerLevel(root->left) - computerLevel(root->right);
if (difference >= -1 && difference <= 1) {
return isBalanced(root->left) && isBalanced(root->right);
}
else {
return false;
}
}
private:
int computerLevel(TreeNode* node) {
if (node == nullptr) return 0;
return max(computerLevel(node->left), computerLevel(node->right)) + 1;
}
};