题目描述:
解法思路:
其实这道题和上一道题求二叉树的深度差不多,我们可以定义一个布尔类型的flag,初始值为false,用来判断是否是平衡二叉树。在上道题的基础上,先求出每层左右子树的深度,然后看左右子树的深度的绝对值相差是否大于1。大于1,证明该树不是平衡二叉树,flag改为false。最后返回flag即可。
值得注意的是,我们要注意剪枝的问题,不然就会出现超时的情况。什么是剪枝呢?就好像已经知道左子树存在不符合平衡二叉树的情况,此时就没有必要再去判断右子树了,也就是剪去右子树不做判断。所以我们可以通过flag变量,当flag已经为false,就直接向上返回,不再向下继续判断子树的深度了。
代码实现:
class Solution {
boolean flag = true;
public boolean isBalanced(TreeNode root) {
maxdep(root);
return flag;
}
int maxdep(TreeNode root){
if(root == null || !flag) return 0;
int left = maxdep(root.left);
int right = maxdep(root.right);
if(Math.abs(left - right) > 1) {
flag = false; return 0;
}
return Math.max(left, right) + 1;
}
}
执行结果: