设计一个算法判断一棵树是否为平衡
平衡二叉树:一棵树左右子树高度差不超过1,则为平衡二叉树
- 遍历左、右子树,head为空时为遍历出口,并返回0
- i和j代表返回的值
- 如果i和j的差大于1,则返回-1,若i或j中一个小于1,则返回-1
- 若正常,这返回i和j中较大那个,并加1,代表此层已经遍历完
public class CheckBalance {
public boolean check(TreeNode root) {
// write code here
return m(root)>=0;
}
public int m(TreeNode head){
if(head==null)//判断是否为空节点,是的话返回0,即遍历出口
return 0;
int l= m(head.left);//遍历左子树
int r= m(head.right);//遍历右子树
if(l<0||r<0)//如果存在不含平衡二叉树的子树,则返回-1
return -1;
if(Math.abs(r-l)>1)//如果一个子树两边的深度大于1,则不是平衡二叉树,返回-1
return -1;
return Math.max(l,r)+1;//返回左右子树中深度较大的那个,并加1
}
}