LeetCode Balanced Binary Tree (判断平衡树)

题意：如题，平衡树是指任意一个节点（除了叶子），其左子树的高度与右子树的高度相差不超过1。

思路：递归解决，但是提供的函数不满足递归的要求啊，我们至少得知道高度，又得返回真假，所以另开个函数解决。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool ans;//答案会记录在此。
    int DFS(TreeNode*  root)
    {
        if(!root || !ans)   return 0;  //已经错了，不必要再比较下去
        int b=DFS(root->right);        
        int a=DFS(root->left);
        if(abs(a - b)>1)  ans=false;
        return max(a,b)+1;
    }

    bool isBalanced(TreeNode* root) {
        ans=true;//初始化答案
        DFS(root);
        return ans;
    }
};