剑指offer 平衡二叉树

题目描述：

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

思路：在得到树的高度之后，对左右树的高度进行判断，如果绝对值之差大于1则不是平衡二叉树。

代码：

class Solution 
{
public:
    bool ok = true;
    int abs(int a,int b)
    {
        if(a>b)  return a-b;
        else     return b-a;
    }
    bool IsBalanced_Solution(TreeNode* pRoot) 
    {
        if(!pRoot) return true;
        int ans = dfs(pRoot);
        return ok;
    }
    int dfs(TreeNode* pRoot)
    {
        if(!pRoot) return 0;
        int left_depth = dfs(pRoot->left);
        int right_depth = dfs(pRoot->right);
     	if(abs(left_depth,right_depth)>1)  ok = false;
        return left_depth>right_depth?left_depth+1:right_depth+1;
    }
};