题目描述
输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。
输入输出样例
给定二叉树 [3,9,20,null,null,15,7]
3
/ \
9 20
/ \
15 7
返回 true 。
题解
题目本质上是比较左右子树的高度差
- 使用后序遍历得到左右子树的高度
- 由于题目给定的方法返回值是boolean类型的,所以需要使用一个辅函数
- 如果子树中有非平衡子树的存在,就返回不可能出现的数字(-1)
class Solution {
public boolean isBalanced(TreeNode root) {
// 如果等于-1说明子树中出现了非平衡树,返回false
return helper(root) != -1;
}
public int helper(TreeNode root){
if(root == null) return 0;
int left = helper(root.left);
int right = helper(root.right);
if(left == -1 || right == -1) return -1;
if(Math.abs(left-right) > 1) return -1;
return Math.max(left, right) + 1;
}
}