二叉树平衡检查LeetCode简单题

平衡的定义如下，已知对于树中的任意一个结点，若其两颗子树的高度差不超过1，则我们称该树平衡。现给定指向树根结点的指针TreeNode* root，请编写函数返回一个bool，表示该二叉树是否平衡。

方法一

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/

class Balance {
public:
    bool isBalance(TreeNode* root) {
        // write code here
        if(root == 0) return true;
        int ld = getdepth(root->left);
        int rd = getdepth(root->right);
        if(abs(ld - rd) > 1) return false;
        return isBalance(root->left) && isBalance(root->right);
    }
    int getdepth(TreeNode* root){
        return root ? 1+max(getdepth(root->left), getdepth(root->right)) : 0;
    }
};

方法二

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/

class Balance {
public:
    bool isBalance(TreeNode* root) {
        // write code here
        dfs(root, 0);
        return flag;
    }
    int dfs(TreeNode* root, int depth){
        if(root == 0) return depth;
        int ld = dfs(root->left, depth+1);
        int rd = dfs(root->right, depth+1);
        if(abs(ld - rd) > 1) flag = 0;
        return max(ld, rd);
    }
    bool flag = 1;
};