输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1,那么它就是一棵平衡二叉树。
bool IsBalanced(TreeNode *root, int &depth) {
if (nullptr == root)
return true;
int left = 0, right = 0;
if (IsBalanced(root->left, left) && IsBalanced(root->right, right)) {
depth = left > right ? left + 1 : right + 1;
int diff = abs(left - right);
if (diff > 1)
return false;
return true;
}
return false;
}
bool isBalanced(TreeNode* root) {
int depth = 0;
return IsBalanced(root, depth);
}