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 };