剑指 Offer 55 - II. 平衡二叉树(DFS)

解法：树的深度遍历

先做这道题：剑指 Offer 55 - I. 二叉树的深度（递归）

自底向上，比较每个节点左子树与右子树的高，相差不大于1，则为平衡二叉树。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */
int height(struct TreeNode* root, bool *flag)
{
    if(!root) return 0;
    int left = height(root->left, flag);
    int right = height(root->right, flag);
    if(abs(left - right) > 1) *flag = false;
    return left > right ? ++left : ++right;
}

bool isBalanced(struct TreeNode* root){
    bool flag = true;
    height(root, &flag);
    return flag;
}

时间复杂度$O(n)$:会递归遍历每个节点
空间复杂度$O(n)$:最坏情况下，当二叉树退化为一条链表时，系统栈深度为n。