题目: 输入一棵二叉树的根节点,判断该树是不是平衡二叉树。
分析: 平衡二叉树(Balance Tree)的定义是:二叉树中任意节点的左右子树的深度相差不超过1。注意是任意节点,并不只是根节点的|左深度-右深度|
,还有其子树也必须是平衡二叉树。从该定义也可知是利用递归来解决这个问题。
代码:
class Solution {
public boolean isBalanced(TreeNode root) {
if(root==null) return true;
return Math.abs(depth(root.left)-depth(root.right))<=1 && isBalanced(root.left) && isBalanced(root.right);
}
private int depth(TreeNode node) {
if(node==null) return 0;
return Math.max(depth(node.left), depth(node.right)) + 1;
}
}
运行结果:
解析:
-
代码思路是:若树为空,则为BT;若为非空二叉树,则求出左、右子树的深度depth,若
|左深度-右深度|<=1
,同时其左子树和右子树均为BT,则该树为BT。 -
时间复杂度:O(N logN),最差情况下(为 “满二叉树” 时),求树的深度为
logN
,而这里每个子树都需要求其深度,所以为N*logN
。 -
空间复杂度:O(N),最差的情况下树退化成链表,需要O(N)的栈空间。
-
但此方法需要遍历节点多次,那有没有每个节点只遍历一次的解法呢?——我把
return Math.abs(depth(root.left)-depth(root.right))<=1 && isBalanced(root.left) && isBalanced(root.right);
改成了return isBalanced(root.left) && isBalanced(root.right) && Math.abs(depth(root.left)-depth(root.right))<=1
,执行结果是错误的,分析后才发现,&&
是短路逻辑与,在执行本句代码时,会先计算isBalanced(root.left)
,这样递归下去直到叶子节点的子节点是null,会返回true,而这本身就是错误的逻辑。 -
题解中的方法是利用从底至上的方法,后序遍历+剪枝
public boolean isBalanced(TreeNode root) {
int check = check(root);
if(check<0) return false;
else return true;
}
private int check(TreeNode root) {
if(root==null) return 0;
int l_depth = check(root.left);
if(l_depth==-1) return -1;
int r_depth = check(root.right);
if(r_depth==-1) return -1;
if(Math.abs(l_depth-r_depth)>1) return -1;
else return Math.max(l_depth,r_depth)+1;
}
思路详解:当节点root 的左 / 右子树的深度差 ≤1 :则返回当前子树的深度,即节点 root 的左 / 右子树的深度最大值 +1;当节点root 左 / 右子树的深度差 > 1 :则返回 -1,代表此子树不是平衡树。