剑指No.55-2_平衡二叉树
- 题目: 输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。
示例:
3
/ \
9 20
/ \
15 7 返回ture
- 后序遍历+剪枝
public class MySolution {
public boolean isBalancedWay(TreeNode root) {
return recur(root) != -1;
}
public int recur(TreeNode root){
// 当节点为空时,此节点的深度为0
if (root == null)
return 0;
// 创建left和right分别保存左右子树的深度
int left = recur(root.leftNode);
// 如果出现了左右子树深度大于1的情况,就会返回-1,此时不用再看其他直接剪枝向上返回-1
if (left == -1)
return -1;
int right = recur(root.rightNode);
if (right == -1)
return -1;
// 查看左右子树的深度差是不是小于2,如果到了2就说明不是平衡二叉树,返回-1
// 如果没到2,说明还是平衡二叉树,向上返回此子树的深度,也就是当前节点中左右子树最大深度再加上当前节点的1
return Math.abs(left - right) < 2 ? Math.max(left, right) + 1 : -1;
}
}