【力扣面试】面试题 04.04. 检查平衡性
题目
实现一个函数,检查二叉树是否平衡。在这个问题中,平衡树的定义如下:任意一个节点,其两棵子树的高度差不超过 1。
示例 1:
示例 2:
解题思路
1、分别求取当前节点的左子树高度和右子树高度,比较其差值是否小于1
2、并且判断当前节点的左孩子节点是否满足1条件
3、并且判断当前节点的右孩子节点是否满足1条件
4、都满足的话,即可返回true
代码
class Solution {
public boolean isBalanced(TreeNode root) {
if (root==null){
return true;
}
//分别计算左子树和右子树的高度
int left = getDepth(root.left);
int right = getDepth(root.right);
//这两个子树的高度不能超过1,并且他的两个子树也必须是平衡二叉树
return Math.abs(left-right) <= 1 && isBalanced(root.left) && isBalanced(root.right);
}
//计算树中节点的高度
public int getDepth(TreeNode root){
if (root == null) {
return 0;
}
return (Math.max(getDepth(root.left), getDepth(root.right)) + 1);
}
}