平衡二叉树计算高度的同时判断是否平衡

最新推荐文章于 2022-12-09 09:28:46 发布

原创最新推荐文章于 2022-12-09 09:28:46 发布 · 1.6k 阅读

0 ·

CC 4.0 BY-SA版权

算法同时被 2 个专栏收录

45 篇文章

订阅专栏

笔试面试

37 篇文章

订阅专栏

本文详细介绍了如何解答微软电话面试中关于二叉树平衡性的代码问题，通过给出一个示例代码，解释了如何判断一个二叉树是否平衡，并讨论了代码实现中的关键点。

今天一个朋友微软电话面试，问到一个问题，就是如题所述的，他问我他写的有没有错，挺难说的，然后我就随手写了一个版本（自我感觉良好）～

bool balance(Node* root, int& height) {
    if (root == NULL) {
        height = 0;
        return true;
    }

    int left_height = 0, right_height = 0;
    bool left_b = balance(root->left, left_height);
    bool right_b = balance(root->right, right_height);

    height = max(left_height, right_height) + 1;
    if (left_height - right_height > 1 || right_height - left_height > 1)
        return false;
    return left_b && right_b;
}