问题描述:
输入一棵二叉树的根结点,判断该树是不是平衡二叉树。如果某二叉树中任意结点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。
注意:
规定空树也是一棵平衡二叉树。
样例:
只要二叉树中任意左右子树不满足高度相差为 1 就返回结果,所以设置全局变量ans,如果整棵树没有判断完就出现了不满足的情况,那么立马返回无需再进行后序判断,其中函数dfs用于记录树的高度,在记录的过程中判断此时的左右子树高度是否满足条件。
class Solution {
public:
bool ans=true;
bool isBalanced(TreeNode* root) {
dfs(root);
return ans;
}
int dfs(TreeNode* root)
{
if(!root) return false;
int left=dfs(root->left),right=dfs(root->right);
if(abs(left-right)>1) ans=false;
return max(left,right)+1;
}
};