输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。
示例 1:
给定二叉树 [3,9,20,null,null,15,7]
3
/
9 20
/
15 7
返回 true
解题思路
树的深度depth(root) = max(depth(root.left), depth(root.right)) + 1
是否平衡 = 左右子树深度差 < 2 且 左子树是平衡的 且 右子树是平衡的
代码
class Solution {
public int maxDepth(TreeNode root) {
if (root == null) {
return 0;
}
// depth(root) = max(depth(root.left), depth(root.right)) + 1
return Math.max(maxDepth(root.left), maxDepth(root.right)) + 1;
}
public boolean isBalanced(TreeNode root) {
return root == null || Math.abs(maxDepth(root.left) - maxDepth(root.right)) < 2 && isBalanced(root.left) && isBalanced(root.right);
}
}