思路
树的问题都用递归做就好了。递归需要看:这个递归是在何处停止的,返回的应该是什么。
比如这道题,题目当中已经给出了,平衡二叉树的定义就是,左子树和柚子树的高度差不大于1的树。所以很自然的可以想到通过求两边子树的高度差来判断。那么怎么求高度差呢,就需要引入maxdepth函数,这个函数的写法在上一篇已经写过了,也算是趁热打铁拿过来用一下。如果高度差>1则返回false,如果输入的为空则返回true,因为如果能走到这一步就说明上面的都是平衡的了。
代码
bool isBalanced(TreeNode* root) {
if(root==NULL){return true;}
if(maxdepth(root->left)-maxdepth(root->right)>1||maxdepth(root->right)-maxdepth(root->left)>1){return false;}
return isBalanced(root->left)&&isBalanced(root->right);
}
int maxdepth(TreeNode* p){
if(p==NULL){
return 0;
}
return max(maxdepth(p->left),maxdepth(p->right))+1;
}
运行结果
还不错~