剑指offer 平衡二叉树（C++）

题目描述

输入一棵二叉树，判断该二叉树是否是平衡二叉树。

解题思路

1、空树为平衡二叉树

2、二叉树左右子树的高度差绝对值不超过1为平衡二叉树，该树的左右子树也满足这个条件，即递归问题。

代码实现

class Solution {
public:
    int GetTreeLength(TreeNode* pRoot) // 求二叉树的高度
    {
        if(pRoot == NULL)
            return 0;
        int leftLen = GetTreeLength(pRoot->left);
        int rightLen = GetTreeLength(pRoot->right);
        return max(leftLen, rightLen) + 1;
    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
        if(pRoot == NULL)
            return true;
        if(abs(GetTreeLength(pRoot->left) - GetTreeLength(pRoot->right)) > 1)
            return false;
        bool leftBal = true, rightBal = true;
        leftBal = IsBalanced_Solution(pRoot->left);
        rightBal = IsBalanced_Solution(pRoot->right);
        return leftBal && rightBal;
    }
};