LeetCode 110. 平衡二叉树（二叉树高度）

示例 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 。

限制：
1 <= 树的结点个数 <= 10000

class Solution {
public:
    bool isBalanced(TreeNode* root) {
        bool bal = true;
        balence(root,bal);
        return bal;
    }
    int balence(TreeNode* root, bool &bal) 
    {
    	if(!bal)
    		return -1;
    	if(root == NULL)
    		return 0;
    	int left_height = balence(root->left,bal);
    	int right_height = balence(root->right,bal);
    	if(abs(left_height-right_height)>1)
    		bal = false;
    	return max(left_height,right_height)+1;
    }
};

class Solution {
	bool balanced = true;
public:
    bool isBalanced(TreeNode* root) {
        depth(root);
        return balanced;
    }

    int depth(TreeNode* root)
    {
    	if(!balanced || !root)
    		return 0;
    	int depthL = depth(root->left);
    	int depthR = depth(root->right);
    	if(abs(depthL - depthR) > 1)
    		balanced = false;
    	return 1+max(depthL, depthR);
    }
};