原来是之前做过的题目“判断二叉树深度“的拓展题:
bfs&&dfs&&剑指 Offer 55 - I. 二叉树的深度
返回值是bool类型时,return 的代码不一定直接写 return true or return false,也有返回一个判断和条件来让机器自己判断是否是true or false。
话不多说,参考了K神的题解:
面试题55 - II. 平衡二叉树(从底至顶、从顶至底,清晰图解)
1.先序遍历+判断深度(感觉是从avl树的定义入手,但是有重复计算的部分
avl树:(1)空树;(2)左右子树的深度差绝对值不超过1,且左右子树也是avl树。
/**
* 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:
int depth(TreeNode*root){
if(!root) return 0;
return max(depth(root->left),depth(root->right))+1;
}
bool isBalanced(TreeNode* root) {
if(!root) return true;
return abs(depth(root->left)-depth(root->right))<=1 && isBalanced(root->left) && isBalanced(root->right);
}
};
2.后序遍历+剪枝(剪枝操作省去了大部分的重复操作
/**
* 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:
int depth(TreeNode*root){
if(!root) return 0;
int left=depth(root->left);
if(left==-1) return -1;
int right=depth(root->right);
if(right==-1) return -1;
return abs(left-right)<=1 ? max(depth(root->left),depth(root->right))+1 : -1;
}
bool isBalanced(TreeNode* root) {
return depth(root)!=-1;
}
};