【二叉树 Java】判断二叉树是否为高度平衡二叉树

一、题目描述

给定一个二叉树，判断它是否是高度平衡的二叉树。本题中，一棵高度平衡二叉树定义为：一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。

示例：

二、思路

递归的去看其左右子树是否是高度平衡二叉树，递归退出条件为当前节点为 null ， 返回值为是否是高度平衡二叉树。

本题可以这样思考 —— 我们想要对每个节点判断一遍其左右子树高度差是否小于或等于1，因此完成本题需要如下几步：

• 第一步： 辅助函数 —— 辅助函数是一求高度的递归函数，用于求出左右子树的高度。
• 第二步： 主要函数 —— 判断是否是高度平衡二叉树的函数
  —— 递归地对二叉树中的每个节点的左右子树用辅助函数求出高度并进行高度差是否小于或等于一的判断。
  （1）若高度差不小于等于 1 ， 返回 false，否则返回 true 。
  （2）若返回了 false 可直接判断递归停止，结果即为 false，若返回 true ， 则还要递归其他节点继续判断其他节点是否符合平衡二叉树的条件。

三、代码

    // 判断二叉树是否为高度平衡二叉树 —— 判断左右子树是否是高度平衡二叉树
    public boolean isBalanced(TreeNode root) {
        if(root == null){ // 空树是高度平衡二叉树
            return true;
        }
        if(!isBalanced(root.left) || !isBalanced(root.right)){ // 左右子树不是二叉平衡树，就不是二叉平衡树
            return false;
        }
        return Math.abs(depth(root.left) - depth(root.right)) <= 1; // 高度差最多为1
    }

    // 以当前节点为根节点的树的深度
    public int depth(TreeNode node){
        if(node == null){
            return 0;
        }
        return Math.max(depth(node.left), depth(node.right)) + 1;
    }