【JZ79 判断是不是平衡二叉树】

描述

输入一棵节点数为 n 二叉树，判断该二叉树是否是平衡二叉树。
在这里，我们只需要考虑其平衡性，不需要考虑其是不是排序二叉树
平衡二叉树（Balanced Binary Tree），具有以下性质：它是一棵空树或它的左右两个子树的高度差的绝对值不超过1，并且左右两个子树都是一棵平衡二叉树。
样例解释：

样例二叉树如图，为一颗平衡二叉树
注：我们约定空树是平衡二叉树。

数据范围： n ≤ 100 ,树上节点的val值满足 0 ≤ n ≤ 1000
要求： 空间复杂度O(1)O(1)，时间复杂度 O(n)O(n)

输入描述：

输入一棵二叉树的根节点

返回值描述：

输出一个布尔类型的值

示例1

输入： {1,2,3,4,5,6,7}

返回值：true

示例2

输入：{}

返回值：true

解法一：自顶向下

一棵二叉树为「平衡二叉树」的条件为：该树为空树，或者其左右子树的高度差最大为1。因此，判断一棵二叉树是否平衡需要求其子树高度，并比较左右子树高度差。

因此此题的解题步骤如下：

1. 设计「求二叉树高度」的函数getHeight(root)，作用是求以root为根结点的二叉树高度；
2. 遍历原二叉树，对其每个结点调用getHeight(root)函数，若存在某左右子树的高度差大于等于2，则是不平衡的；否则是平衡二叉树。

求取二叉树高度的思路如图所示。

图中，红色箭头方向为递归方向，绿色箭头方向为回溯方向。

1. 当结点为空时，其高度为0；
2. 当结点无左孩子、右孩子时，其高度为1；
3. 否则，该结点的高度为return (左子树高度，右子树高度) + 1。（加1是因为要加上该结点本身）

class Solution {
public:
    //计算该子树深度 fast-template
    int deep(TreeNode* root){
        //空结点深度为0
        if(root == NULL)
            return 0;
        //递归算左右子树的深度
        int left = deep(root->left);
        int right = deep(root->right);
        //子树最大深度加上自己
        return (left > right) ? left + 1 : right + 1;
    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
         //空树为平衡二叉树
        if (pRoot == NULL)
            return true;
        int left = deep(pRoot->left);
        int right = deep(pRoot->right);
        //左子树深度减去右子树相差绝对值大于1
        if(left - right > 1 || left - right < -1){
            return false;
        }
        //同时，左右子树还必须是平衡的
        return IsBalanced_Solution(pRoot->left) && IsBalanced_Solution(pRoot->right);
    }
};

运行时间：4ms
超过9.49% 用C++提交的代码
占用内存：524KB
超过60.64%用C++提交的代码
在计算一个结点的高度时，平均情况下时间复杂度为O(logN)；且对于每个结点分别计算其高度，故总时间复杂度为O(NlogN)；该算法需要递归地遍历每个结点，因此空间复杂度为O(N)。

解法二：自底向上

解法一在「计算二叉树高度」时遍历了树的结点，在「判断是否平衡」时又遍历了一次树的结点，因此产生了重复计算。

针对解法一的优化是采用「自底向上」的解题思路：

1. 从最底的叶子结点开始，计算该结点的高度。若该结点不平衡，则直接返回-1，不用继续计算其他结点高度，否则返回其高度；
2. 若自底向上的过程中一直都是平衡的，则最终的树是平衡的。

此方法每个结点（最坏时）仅会遍历一次，不会有重复计算。

class Solution {
public:
    bool IsBalanced_Solution(TreeNode* pRoot) {
        if (!pRoot)
            return true;
        return getHeight(pRoot) >= 0; // 若结果不是-1，则是平衡的 
    }
     int getHeight(TreeNode* root) {
        if (!root)
            return 0;
        // 左子树高度
        int left = getHeight(root->left);
        if (left == -1)
            return -1; // 若左子树高度为-1，则不平衡
        int right = getHeight(root->right); // 右子树高度
        if (right == -1 || abs(left - right) >= 2)
            return -1; // 若右子树高度为-1，或左右子树高度之差大于1，则不平衡
        return 1 + max(left, right); // 返回该结点高度
    }
};

运行时间：3ms
超过32.71% 用C++提交的代码
占用内存：592KB
超过48.96%用C++提交的代码
该方法（最坏情况）访问每个结点一次，因此时间复杂度为O(N)，该方法需要递归遍历每个结点，需要函数调用的开销，空间复杂度为O(N)。