程序员面试金典 4.4

Check Balanced：检查一个二叉树是否是平衡的，平衡定义为一个节点的两颗子树高度最多差1。

根据平衡的定义，可以实现一个递归算法，计算每个节点的两颗子树的高度然后进行判断。

按照朴素的解法，要先判断左右子树是否平衡，然后根据左右子树的高度再判断整棵树是否平衡，这样要进行两次递归，通过优化可以将判断平衡和计算高度合并为一次递归。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool isBalanced(TreeNode* root) {
        if(root == nullptr) return true;
        int lHgt, rHgt;
        bool bLeft = checkBalance(root->left, lHgt);
        bool bRight = checkBalance(root->right, rHgt);
        bool bRoot = abs(lHgt - rHgt) <= 1;
        return bLeft && bRight && bRoot;
    }
private:
    bool checkBalance(TreeNode *root, int &height)
    {
        if(root == nullptr){
            height = -1;
            return true;
        }
        int lHgt, rHgt;
        bool bLeft = checkBalance(root->left, lHgt);
        bool bRight = checkBalance(root->right, rHgt);
        bool bRoot = abs(lHgt - rHgt) <= 1;
        height = lHgt > rHgt ? lHgt : rHgt;
        height++;
        return bLeft && bRight && bRoot; 
    }
};