思路1:先判断左右子树是不是平衡的,若平衡再求出左右子树的深度,若深度之差大于1,则不平衡。贴上代码:
int Depth(TreeNode *root)
{
if (!root) return 0;
return 1 + max(Depth(root->left),Depth(root->right));
}
bool isBalanced(TreeNode *root)
{
if (!root) return true;
if(!isBalanced(root->left)) return false;
if(!isBalanced(root->right)) return false;
int depthLeft = Depth(root->left);
int depthRight = Depth(root->right);
if (abs(depthLeft - depthRight) > 1)
return false;
return true;
}
因为在遍历每个结点时都要求其左右子树的深度,因此复杂度是O(n^2)的。但可以发现,为每个结点计算树的深度是重复的,如左右子树的深度求得的情况下,其直接父亲树的深度就可以不必求了。
思路2:在判断左右子树是否平衡的过程中把深度计算出来,这样在对父结点进行平衡判断时就可以不用再重复计算左右子树的深度了。
bool isBalanced(TreeNode *root, int &height)
{
if(!root) return true;
int left=0,right=0;
if(!isBalanced(root->left,left)) return false;
if(!isBalanced(root->right,right)) return false;
if(abs(left-right) > 1) return false;
height = max(left,right) + 1;
return true;
}
相当于后序遍历,因此复杂度是O(n)的。