leetcode_110. 平衡二叉树

给定一个二叉树，判断它是否是高度平衡的二叉树。

本题中，一棵高度平衡二叉树定义为：

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1。

示例 1:

给定二叉树 [3,9,20,null,null,15,7]

    3
   /   \
  9   20
       /  \
    15    7

返回 true 。

示例 2:

给定二叉树 [1,2,2,3,3,null,null,4,4]
            1
          /   \
        2     2
      /   \
    3     3
   /  \
 4    4

返回 false 。

解析：本题要求判断给定的树是否高度平衡。
对于每一个节点我们需要做以下4步：
1、获取左节点的深度
2、获取右节点的深度
3、判断左右节点相差是否超过1
4、不超过1返回该节点的深度

从逻辑去想我们或许需要两种递归函数，一种用来获取深度，另一种用来判断高度是否平衡，但是这样递归次数太多太慢
我们可以利用负数，对于一个节点的深度，它最小为0，即空节点，因此我们可以定义如果节点的深度为负数则代表该节点高度不平衡，这样就可以获取深度的同时判断是否高度平衡

class Solution {
public:
    bool isBalanced(TreeNode* root) {
        return (nodeDepth(root)!=-1);
    }
    //判断该节点以及其左右子树各节点是否高度平衡，不平衡返回-1
    //平衡返回该节点深度
    int nodeDepth(TreeNode* root){
        if(!root) return 0; //空节点高度为0
        if(!root->left&&!root->right) return 1; //叶子节点高度为1
        //左右子树不符合要求层层返回
        int l = nodeDepth(root->left);
        int r = nodeDepth(root->right);
        if(l<0) return -1; 
        if(r<0) return -1;
        if(abs(l-r)>1) return -1;
        //均符合要求的情况下返回该节点的深度
        return max(l,r)+1;
    }
};