LeetCode Balanced Binary Tree (判断平衡树)

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

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

 1 /**
 2  * Definition for a binary tree node.
 3  * struct TreeNode {
 4  *     int val;
 5  *     TreeNode *left;
 6  *     TreeNode *right;
 7  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 8  * };
 9  */
10 class Solution {
11 public:
12     bool ans;//答案会记录在此。
13     int DFS(TreeNode*  root)
14     {
15         if(!root || !ans)   return 0;  //已经错了，不必要再比较下去
16         int b=DFS(root->right);
17         int a=DFS(root->left);
18         if(abs(a - b)>1)  ans=false;
19         return max(a,b)+1;
20     }
21
22     bool isBalanced(TreeNode* root) {
23         ans=true;//初始化答案
24         DFS(root);
25         return ans;
26     }
27 };

AC代码