面试题55（二）：平衡二叉树

题目描述
输入一棵二叉树，判断该二叉树是否是平衡二叉树。
方法一：比较各节点深度(存在重复遍历节点情况)。

class Solution {
public:
    int getDepth(TreeNode* root)
    {
        if(root == nullptr)
            return 0;
        int leftH = getDepth(root->left);
        int rightH = getDepth(root->right);
        return max(leftH, rightH) + 1;
    }

    bool IsBalanced_Solution(TreeNode* pRoot) {
        if(pRoot == nullptr)
            return true;
        int left = getDepth(pRoot->left);
        int right = getDepth(pRoot->right);
        int dif = left - right;
        if(dif > 1 || dif < -1)
            return false;
        return IsBalanced_Solution(pRoot->left) && IsBalanced_Solution(pRoot->right);
    }
};

方法二：使用后续遍历方式遍历二叉树的每个节点，在遍历到一个节点之前我们就已经遍历了它的左右子树，则不存在重复遍历。

class Solution {
public:
    bool isBalanced(TreeNode* root, int& depth)
    {
        if(root == nullptr)
        {
            //depth = 0;
            return true;
        }
        int left = 0, right = 0;
        if(isBalanced(root->left, left) && isBalanced(root->right, right))
        {
            int dif = left - right;
            if(dif <= 1 && dif >= -1)
            {
                depth = max(left, right) + 1;
                return true;
            }
        }
        return false;
    }

    bool IsBalanced_Solution(TreeNode* pRoot) {
        int depth = 0;
        return isBalanced(pRoot, depth);
    }
};