二叉树平衡检查DFS

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

class Balance {
public:
    int flag = 0;
    int dfs(TreeNode* root, int deepth) {
        if (root == NULL) {
            return deepth;
        }
        
        int ldeep = dfs(root->left, deepth + 1);
        int rdeep = dfs(root->right, deepth + 1);
        
        if (abs(ldeep - rdeep) > 1) {
            flag = 1;
        }
        
        return max(ldeep, rdeep);
    }
    bool isBalance(TreeNode* root) {
        dfs(root, 0);
        return flag == 0;
    }
};