根据题意,我们只需要比较当前节点的左右子树高度差是否小于1,利用分治法,只需要满足:
1. 根节点的左右子树的高度差小于1。
2. 根节点左右子树的满足高度差小于1,在往下走,判断左子树根节点的左右子树是否满足,右子树根节点的左右子树是否满足。
int TreeDepth(struct TreeNode* root)
{
if(root == NULL)
{
return 0;
}
int left = TreeDepth(root->left)+1;
int right = TreeDepth(root->right)+1;
return left>right?left:right;
}
bool isBalanced(struct TreeNode* root) {
if(root == NULL)
{
return true;
}
int left_depth = TreeDepth(root->left);
int right_depth = TreeDepth(root->right);
if(abs(left_depth-right_depth) > 1)
{
return false;
}
return isBalanced(root->left)&&isBalanced(root->right);
}