平衡二叉树的定义是递归的:既平衡二叉树的左子树和右子树也是平衡二叉树;当左右子树的高度相差不大于1时,就说这个 二叉树是平衡的。
二叉树的节点定义如下:
public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
判断二叉树是否为平衡二叉树的思路:首先计算机左右子树的高度,判断高度差是否小于1,递归判断左子树与右子树;如果所有的高度差都小于1,则这棵树是平衡的,代码如下:(返回-1表示这棵树是不平衡的)
public class Solution {
public boolean isBalanced(TreeNode root) {
if(null==root){
return true;
}
return depthOfTree(root)!=-1;
}
public int depthOfTree(TreeNode root){
if(null==root){
return 0;
}
int leftDepth=depthOfTree(root.left);
int rightDepth=depthOfTree(root.right);
if(-1==leftDepth||-1==rightDepth){
return -1;
}
if(leftDepth-rightDepth>1||rightDepth-leftDepth>1){
return -1;
}
if(leftDepth>rightDepth){
return leftDepth+1;
}else{
return rightDepth+1;
}
}
}