根据题意,我们只需要比较当前节点的左右子树高度差是否小于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);
}