剑指Offer：[第18天 搜索与回溯算法(中等)]---＞Ⅱ. 平衡二叉树

---

一、题目描述

输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过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 

限制：

0 <= 树的结点个数 <= 10000

二、使用步骤

注：思路分析中的一些内容和图片参考自力扣各位前辈的题解，感谢他们的无私奉献

思路：

此题为剑指Offer：[第18天 搜索与回溯算法(中等)]—＞Ⅰ. 二叉树的深度的拓展
可以对二叉树做后序遍历，从底至顶返回子树深度，若判定某子树不是平衡树则剪枝，直接向上返回
算法流程：
recur(root) 函数
①返回值：
当节点root的左右子树的深度差≤1：则返回当前子树的深度，即节点root的左右子树的深度最大值+1，即max(left,right)+1
当节点root的左右子树的深度差>2：则返回−1，代表此子树不是平衡树
②终止条件：
当root为空：说明越过叶节点，因此返回高度0
当左/右子树深度为−1：代表此树的左/右子树不是平衡树，因此剪枝，直接返回−1
isBalanced(root) 函数
①返回值：若recur(root)!=-1，则说明此树平衡，返回true，否则返回false
案例分析：










复杂度分析：
时间复杂度$\mathrm{O}(\mathrm{N})$：N为树的节点数，最差情况下，需要递归遍历树的所有节点
空间复杂度$\mathrm{O}(\mathrm{N})$：最差情况下(树退化为链表时)，系统递归需要使用O(N)的栈空间

三、整体代码

整体代码如下

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */

int height(struct TreeNode* root) {
    //如果为空结点，代表穿过叶节点，深度为0
    if (root == NULL) {
        return 0;
    }
    //递归处理左右子树
    int leftHeight = height(root->left);  
    int rightHeight = height(root->right);
    //如果左子树不平衡  或者右子树不平衡  或者左右子树深度差大于1
    //则代表当前结点不平衡，直接剪枝，返回-1
    if (leftHeight == -1 || rightHeight == -1 || fabs(leftHeight - rightHeight) > 1) {
        return -1;
    } else {
        //否则返回当前这棵树的深度
        return fmax(leftHeight, rightHeight) + 1;
    }
}

bool isBalanced(struct TreeNode* root) {
    return height(root) >= 0;
}

运行，测试通过