二叉树(判断是否为平衡二叉树)

题目(力扣):


二叉树(判断是否为平衡二叉树)_第1张图片

观察题目,发现最重要的条件就是,两颗子树的高度差的绝对值不超过1,我们就可以用递归将所有左子树和右子树都遍历一个,求出他们的高度差,若差值 > 1,则返回false,否则则继续遍历。

再次之前,写一个辅助函数会使你的代码更简洁而且条理也更清晰:

TreeHeight:

int TreeHeight(struct TreeNode* root)
{
    if(!root)
        return 0;
    int left = TreeHeight(root->left);
    int right = TreeHeight(root->right);
    return left > right ? left + 1 : right + 1;
}

接着就可以按照上面分析的思路走,很快就能解决:

bool isBalanced(struct TreeNode* root) {
    if(!root)
        return true;
    int sub = abs(TreeHeight(root->left)-TreeHeight(root->right));
    if(sub > 1)
        return false;
    return isBalanced(root->left) && isBalanced(root->right);
}

提交结果:


二叉树(判断是否为平衡二叉树)_第2张图片

完整代码:

int TreeHeight(struct TreeNode* root)
{
    if(!root)
        return 0;
    int left = TreeHeight(root->left);
    int right = TreeHeight(root->right);
    return left > right ? left + 1 : right + 1;
}
bool isBalanced(struct TreeNode* root) {
    if(!root)
        return true;
    int sub = abs(TreeHeight(root->left)-TreeHeight(root->right));
    if(sub > 1)
        return false;
    return isBalanced(root->left) && isBalanced(root->right);
}

你可能感兴趣的:(OJ及其经典习题,数据结构,算法,c语言)