【Leetcode】【Easy】Balanced Binary Tree

Given a binary tree, determine if it is height-balanced.

For this problem, a height-balanced binary tree is defined as a binary tree in which the depth of the two subtrees of every node never differ by more than 1.

错误的思路（o(N2)时间复杂度）：

写一个函数a，用递归遍历的方法，用于计算当前结点的高。

主函数从根节点开始，调用a计算其左子结点和右子结点高差值（×N），如果大于1则返回false，如果小于等于1则继续以左子结点和右子节点分别为根（×N），测试其平衡性；

正确的思路（o(N)时间复杂度）：

错误的思路没有正确理解递归。判断平衡二叉树的条件是：①左子树是平衡的；②右子树是平衡的；③左子树和右子树的深度相差不超过1；

因此每一层递归只需要做两件事，判断左右子树是否平衡，判断左右子树深度差。

这样一来，遍历一遍即可获得判定结果。

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool isBalanced(TreeNode *root) {
        int dep = 0;
        return checkBalance(root, &dep);
    }
    
    bool checkBalance(TreeNode *node, int *dep) {
        if (node == NULL) 
            return true;
        
        int leftDep = 0;
        int rightDep = 0;
        if (checkBalance(node->left, &leftDep) && 
            checkBalance(node->right, &rightDep) && 
            (abs(rightDep - leftDep) <= 1)) {
            *dep = max(leftDep, rightDep) + 1;
            return true;
        } else { 
            return false;
        }
    }
};