判断给定的二叉树是否是平衡的
在这个问题中,定义平衡二叉树为每个节点的左右两个子树高度差的绝对值不超过1的二叉树
解题思路:
每个节点的高度为最大子树高度+1,因此可以用递归的思想计算每个结点的左右子树大小。设置一个标志位flag=0,当计算出一个节点的左右子树大小之后进行比较,如果相差超过1则将标志位设为flag=-1。如果flag=-1,则说明不是平衡二叉树,如果flag=0则说明是平衡二叉树。
class Solution {
public:
int flag=0;
bool isBalanced(TreeNode *root) {
if(!root)
return true;
int res=maxDepth(root);
if(flag==-1)return false;
else return true;
}
int maxDepth(TreeNode *root) {
if(root == NULL) return 0;
int l = maxDepth(root->left);//左子树深度
int r = maxDepth(root->right);//右子树深度
if((l-r)>1||(l-r)<-1) flag=-1;
return max(l,r) + 1;
}
};
缺点:不能在判断为非平衡二叉树之后直接跳出循环,会带来更多的计算。