【leetcode】Balanced Binary Tree（middle）

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.

思路：

我居然在这道题上卡了一个小时。关键是对于平衡的定义，我开始理解错了，我以为是要所有叶节点的高度差不大于1.

但题目中的定义是如下这样的：

Below is a representation of the tree input: {1,2,2,3,3,3,3,4,4,4,4,4,4,#,#,5,5}:

        ____1____
       /               2           2
     /  \        /     3    3      3   3
   /\    /\    /  4  4  4  4  4  4
 /5  5

Let‘s start with the root node (1). As you can see, left subtree‘s depth is 5, while right subtree‘s depth is 4. Therefore, the condition for a height-balanced binary tree holds for the root node. We continue the same comparison recursively for both left and right subtree, and we conclude that this is indeed a balanced binary tree.

我AC的代码：我觉得我的代码就挺好挺短的。

bool isBalanced3(TreeNode* root){
        int depth = 0;
        return isBalancedDepth(root, depth);
    }

    bool isBalancedDepth(TreeNode* root, int &depth)
    {
        if(root == NULL) return true;
        int depthl = 0, depthr = 0;
        bool ans = isBalancedDepth(root->left, depthl) && isBalancedDepth(root->right, depthr) && abs(depthl - depthr) < 2;
        depth = ((depthl > depthr) ? depthl : depthr) + 1;
        return ans;
    }

其他人的代码，对高度多遍历了一遍，会比较慢：

bool isBalanced(TreeNode *root) {
        if (!root) return true;
        if (abs(depth(root->left) - depth(root->right)) > 1) return false;
        return isBalanced(root->left) && isBalanced(root->right);
    }
    int depth(TreeNode *node){
      if (!node) return 0;
      return max(depth(node->left) + 1, depth(node->right) + 1);
    }