题目:
Given a binary tree, determine if it is height-balanced.
For this problem, a height-balanced binary tree is defined as a binary tree in which the depth of the two subtrees of every node never differ by more than 1.
分析:
从二叉树的底部的节点开始检查是否平衡。二叉树的问题大部分可以用分治的思想,即一个节点如果平衡需要它的左右节点都平衡,并且它自己也需要满足条件。但是leetcode的测试用例中有一个很长的单链数据会超时。所以加一个控制,如果某一个节点的某个孩子是null,那么找另一个孩子的高度时不需要递归很深,只要看高度能不能达到2就可以。
Java代码实现:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public boolean isBalanced(TreeNode root) {
if(root==null)
return true;
boolean flag = false;
if(root.left==null || root.right==null)
flag = true;
return isBalanced(root.left) && isBalanced(root.right) && Math.abs((height(root.left, flag)-height(root.right, flag)))<=1;
}
private int height(TreeNode root, boolean flag)
{
if(root==null)
return 0;
if(flag)
{
if(root.left!=null || root.right!=null)
return 2;
}
return Math.max(height(root.left, flag), height(root.right, flag))+1;
}
}