给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1
class Solution {
public boolean isBalanced(TreeNode root) {
if(root == null){
return true;
}
return dfs(root);
}
boolean dfs(TreeNode node){
if(node == null){
return true;
}
if(!isB(node)){
return false;
}
return dfs(node.right) && dfs(node.left);
}
boolean isB(TreeNode root){
if(root == null){
return true;
}
if(root.left == null && root.right == null){
return true;
}
return Math.abs(length(root.left, 1) - length(root.right, 1)) < 2;
}
int length(TreeNode root, int depth){
if(root == null){
return depth;
}
int left = 1, right = 1;
if(root.left == null && root.right == null){
return depth + 1;
}
if(root.left != null){
left = length(root.left, depth + 1);
}
if(root.right != null){
right = length(root.right, depth + 1);
}
depth = Math.max(left, right);
return depth;
}
}