【剑指offer-C++】JZ79：判断是不是平衡二叉树

题目描述

描述：输入一棵节点数为 n 二叉树，判断该二叉树是否是平衡二叉树。

在这里，我们只需要考虑其平衡性，不需要考虑其是不是排序二叉树。

平衡二叉树（Balanced Binary Tree），具有以下性质：它是一棵空树或它的左右两个子树的高度差的绝对值不超过1，并且左右两个子树都是一棵平衡二叉树。

样例解释：

样例二叉树如图，为一颗平衡二叉树。

注：我们约定空树是平衡二叉树。

数据范围：n≤100, 树上节点的val值满足 0≤n≤1000。

要求：空间复杂度O(1)，时间复杂度 O(n)。

输入描述：输入一棵二叉树的根节点。

返回值描述：输出一个布尔类型的值。

输入：{1,2,3,4,5,6,7}
返回值：true

输入：{}
返回值：true

解题思路

判断是不是平衡二叉树：最直观的想法是，首先使用umap存储二叉树结点以及其对应的高度，然后编写一个函数dfs来后序遍历并记忆化搜索存储二叉树各个节点的高度。如果使用递归的思路来考虑该题，则直接考虑平衡二叉树的条件即为左右子树高度差的绝对值不超过1且左右子树均平衡。注意，将判断平衡的部分单独抽离出来并封装成一个函数，否则其会多次调用主函数中的dfs()部分。

抽离前：

    // umap统计 节点 节点高度
    unordered_map<TreeNode *,int> umap;
    // 求高度 后序遍历
    int dfs(TreeNode* cur)
    {
        // 空节点高度为0
        if(!cur)
            return 0;
        // 记忆化搜索
        if(umap.find(cur)!=umap.end())
            return umap[cur];
        // 左
        int left = dfs(cur->left);
        int right = dfs(cur->right);
        // 根节点高度等于左右子树最大高度加一
        return umap[cur]=max(left,right)+1;
    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
        // 默认空树平衡
        if(!pRoot)
            return true;
        dfs(pRoot);
        // 平衡二叉树的要求是左子树平衡且右子树平衡且左右子树高度差小于等于1
        return abs(umap[pRoot->left]-umap[pRoot->right])<=1 && IsBalanced_Solution(pRoot->left) && IsBalanced_Solution(pRoot->right);
    }

抽离后：

    // umap统计 节点 节点高度
    map<TreeNode *,int> umap;
    // 求高度 后序遍历
    int dfs(TreeNode* cur)
    {
        // 空节点高度为0
        if(!cur)
            return 0;
        // 记忆化搜索
        if(umap.find(cur)!=umap.end())
            return umap[cur];
        // 左
        int l = dfs(cur->left);
        int r = dfs(cur->right);
        // 根节点高度等于左右子树最大高度加一
        return umap[cur]=max(l,r)+1;
    }
    bool isBalance(TreeNode* root)
    {
        if(!root)
            return true;
         // 平衡二叉树的要求是左子树平衡且右子树平衡且左右子树高度差小于等于1
        return abs(umap[root->left]-umap[root->right])<=1 && isBalance(root->left) && isBalance(root->right);
    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
        // 默认空树平衡
        if(!pRoot)
            return true;
        dfs(pRoot);
        return isBalance(pRoot);
    }

优化：上述方法我们是先求出高度再判断是否平衡，其实我们可以边求高度边判断平衡。后序遍历求高度，先求左子树高度，再求右子树高度，接着求左右子树高度差绝对值，如果其大于1则表明不平衡则返回-1，反之返回该节点的高度值，同时在左右子树高度返回时也要判断是否为-1，并及时返回-1。

// 求高度 后序遍历
int dfs(TreeNode* cur)
{
   // 空节点高度为0
   if(!cur)
     return 0;
   // 左子树高度
   int left = dfs(cur->left);
   // 左子树平衡则返回高度 反之返回不平衡
   if(left==-1)
      return -1;
   // 右子树高度
   int right = dfs(cur->right);
   // 右子树平衡则返回高度 反之返回不平衡
   if(right==-1)
     return -1;
   // 左右子树高度差
   int diff = abs(left-right);
   // 高度差大于1不平衡
   if(diff>1)
      return -1;
   // 树的高度
   return max(left,right)+1;
}
bool IsBalanced_Solution(TreeNode* pRoot) 
{
    // 默认空树平衡
    if(!pRoot)
        return true;
    return dfs(pRoot)==-1?false:true;
}