在二叉树里有一个概念,当一棵二叉树每个节点 的左右两个子树的高度差的绝对值不超过1时,这棵二叉树成为平衡二叉树,今天讨论怎么判断一棵二叉树是否平衡。
分析:
我们在拿到一棵树的时候,首先看的是根节点,接下来到它的子树,这个时候就应该想办法求它左右子树的深度分别是多少,然后进行比较,如果深度之差的绝对值大于一,那么输出false,否则输出true。
考虑特殊情况:
1.如果这颗树是一颗空树,那么返回true;
2.如果这颗树有且仅有一个根节点,那么返回true;
class Solution {
public int depth(TreeNode root){
if(root == null){
return 0;
}
int left = depth(root.left);
int right = depth(root.right);
return left>right?left+1:right+1;
}
public boolean isBalanced(TreeNode root) {
if(root == null){
return true;
}
boolean leftBalance = isBalanced(root.left);
if(!leftBalance){
return false;
}
boolean rightBalance = isBalanced(root.right);
if(!rightBalance){
return false;
}
int leftdepth = depth(root.left);
int rightdepth = depth(root.right);
int diff = leftdepth-rightdepth;
if(diff>1 || diff<-1){
return false;
}
return true;
}
}
以上便是判断平衡二叉树的代码,但是最后判断的时候有些繁琐,可以用Math.abs加以改进,效果一样,代码更精简,在这里就不写了。