平衡二叉树

题目描述

输入一棵二叉树，判断该二叉树是否是平衡二叉树。

代码

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
public:
    bool IsBalanced_Solution(TreeNode* pRoot) {
        return IsBalanced(pRoot) >= 0;
    }
    
    int IsBalanced(TreeNode* pRoot) {
        if (pRoot == NULL) {
            return 0;
        }
        int l = IsBalanced(pRoot->left), r = IsBalanced(pRoot->right);
        
        if (abs(l - r) <= 1) {//差距为1平衡
            return 1 + max(l, r);
        }
        return -2;//树的高度可以为0,所以返回-2
    }
};