面试题 04.04. 检查平衡性（简单）

题目描述

实现一个函数，检查二叉树是否平衡。在这个问题中，平衡树的定义如下：任意一个节点，其两棵子树的高度差不超过 1。

示例1

给定二叉树 [3,9,20,null,null,15,7]
    3
   / \
  9  20
    /  \
   15   7
返回 true 。

示例2

给定二叉树 [1,2,2,3,3,null,null,4,4]
      1
     / \
    2   2
   / \
  3   3
 / \
4   4
返回 false 。

做题思路

递归计算树的深度，在递归计算左右子树深度的同时判断当前节点所在树是否平衡，即判断左右子树深度差的绝对值是否大于1。

代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    boolean flag=true;
    public boolean isBalanced(TreeNode root) {
        dfs(root);
        return flag;
    }

    //计算树的深度
    public int dfs(TreeNode root){
        if(root==null) return 0;
        int left=dfs(root.left); //左子树深度
        int right=dfs(root.right); //右子树深度
        //判断本结点所在树是否平衡
        if(Math.abs(left-right)>1){
            flag=false;
        }
        //返回本结点所在树的深度（子树的深度+根节点）
        return Math.max(left,right)+1;
    }
}