剑指offer_C++_一天一题_39_平衡二叉树

今日碎碎念:开始看《以家人之名》，太好哭了~

题目描述
输入一棵二叉树，判断该二叉树是否是平衡二叉树。
在这里，我们只需要考虑其平衡性，不需要考虑其是不是排序二叉树
解题分析
平衡二叉树：任意节点的子树的高度差都小于等于1
借助前一题目：计算二叉树深度，遍历每个结点，计算左右子树深度，判断是否平衡。
代码实现
第一代代码：优化前

class Solution {
public:
    bool IsBalanced_Solution(TreeNode* pRoot) {
        if (!pRoot) return true;
        
        unsigned int left = TreeDepth(pRoot->left);
        int right = TreeDepth(pRoot->right);
        bool left_isbalanced = IsBalanced_Solution(pRoot->left);
        bool right_isbalanced = IsBalanced_Solution(pRoot->right);
        
        int d_value=0;
        if(right>left) d_value=right-left;
        else d_value=left-right;

        return d_value<=1 && left_isbalanced && right_isbalanced;
	}
private:
    int TreeDepth(TreeNode* pRoot)
    {
        if (!pRoot) return 0;
        int left = TreeDepth(pRoot->left);
        int right = TreeDepth(pRoot->right);
        
        if (right>left) return right+1;
        else return left+1;
    }
};

第二代代码：优化后
因为前面的结点计算深度和判断的时候有很多重复的地方，所以可以进行优化，某一结点下面一旦有不平衡的现象，整颗就都不平衡了.不进行过多计算了。

class Solution {
public:
    bool IsBalanced_Solution(TreeNode* pRoot) {
        return TreeDepth(pRoot)!=-1;
     }
private:
    int TreeDepth(TreeNode* pRoot)
    {
        if (!pRoot) return 0;
        
        int left = TreeDepth(pRoot->left);
        if (left==-1) return -1;
        int right = TreeDepth(pRoot->right);
        if (right==-1) return -1;
        
        int d_value=0;
        if(right>left) d_value=right-left;
        else d_value=left-right;

        return d_value>1?-1:max(left,right)+1;
    }
};

结果