题目:
输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。
自底向上的递归
由于自顶向下递归,对于同一个节点,会重复遍历同一个节点,导致时间复杂度较高。如果使用自底向上的做法,则对于每个节点,只被遍历一次
自底向上递归的做法类似于后序遍历,对于当前遍历到的节点,先递归地判断其左右子树是否平衡,再判断以当前节点为根的子树是否平衡。如果存在一棵子树不平衡,则整个二叉树一定不平衡。
class Solution {
public:
bool isBalanced(TreeNode *root)
{
int depth = 0;
return isBalanced(root, &depth);
}
bool isBalanced(TreeNode *root, int *depth)
{
if (root == nullptr)
{
*depth = 0;
return true;
}
int left, right;
if (isBalanced(root->left, &left) && isBalanced(root->right, &right))
{
int diff = left - right;
if (diff <= 1 && diff >= -1)
{
*depth = 1 + max(left, right);
return true;
}
}
return false;
}
};
- 时间复杂度:O(n),其中 nn 是二叉树中的节点个数。使用自底向上的递归,每个节点仅遍历一次。
- 空间复杂度:O(n),其中 n 是二叉树中的节点个数。空间复杂度主要取决于递归调用的层数,递归调用的层数不会超过 n。