输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。如给定二叉树 [3,9,20,null,null,15,7],返回true。
力扣把这道题划分为简单,但个人感觉也不是很简单。这里需要计算树的高度,如果高度差在1之内,可以理解为平衡二叉树,所以需要从底向上进行遍历,如果返回的是-1,则意味着不是平衡二叉树。二叉树还是不太好理解,要常常复习。
package leecodeOff;
/*
* 55.平衡二叉树
*/
public class Demo55_2 {
/*
* 判断一棵树是不是平衡二叉树
* 1.如果一棵树是平衡二叉树,则其左、右两个节点的深度差肯定不会超过1
*/
public boolean isBalanced(TreeNode root) {
int value = height(root);
if(value ==-1) {
return false;
}
return true;
}
public int height(TreeNode node) {
if(node == null) {
return 0;
}
int leftheight = height(node.left);
int rightheight = height(node.right);
if(leftheight == -1 || rightheight ==-1 || Math.abs(leftheight-rightheight) >1) {
return -1;
}
return Math.max(leftheight, rightheight)+1;
}
public static void main(String args[]) {
Demo55_2 demo = new Demo55_2();
TreeNode node1 = new TreeNode(3);
TreeNode node21 = new TreeNode(9);
TreeNode node22 = new TreeNode(20);
TreeNode node31 = new TreeNode(15);
TreeNode node32 = new TreeNode(7);
node1.left = node21;
node1.right = node22;
node22.left = node31;
node22.right = node32;
System.out.println(demo.isBalanced(node1));
}
}