数据结构：平衡二叉树（时间复杂度O(N)解法）

最新推荐文章于 2025-02-28 10:57:47 发布

原创最新推荐文章于 2025-02-28 10:57:47 发布 · 8.5k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#平衡二叉树

数据结构专栏收录该内容

9 篇文章

订阅专栏

本文探讨了平衡二叉树的概念，强调其特性为左右子树高度差不超过1。并详细介绍了从最初的O(N^2)时间复杂度到优化后的O(N)时间复杂度解法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

平衡二叉树：一个二叉树每个节点的左右两个子树的高度差的绝对值不超过1

时间复杂度O（N^2）

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */
int TreeHeight(struct TreeNode* root)
{
    if(root == NULL)
    return 0;
    int leftheight = TreeHeight(root->left);
    int rightheight = TreeHeight(root->right);
    return leftheight > rightheight ? leftheight+1 : rightheight+1;
}

bool isBalanced(struct TreeNode* root){
    if(root == NULL)
    return true;
    int leftheight = TreeHeight(root->left);
    int rightheight = TreeHeight(root->right);
    return abs(leftheight-rightheight) < 2 &&
        isBalanced(root->left) &&isBalanced(root->right);
    }

优化：时间复杂度O(N)解法

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */
typedef struct RET
{
    bool isbalanced;
    int height;
}Ret;
Ret _isBalanced(struct TreeNode* root)
{
    Ret ret;
    ret.height = 0;
    ret.isbalanced = true;
    
    if(root == NULL)
        return ret;
    Ret lret;
    lret = _isBalanced(root->left);
    if(lret.isbalanced == false)
    {
        return lret;
    }
    Ret rret;
   rret = _isBalanced(root->right);
    if(rret.isbalanced == false)
    {
        return rret;
    }
    ret.isbalanced = abs(lret.height - rret.height) < 2 ;
    ret.height = lret.height > rret.height ? lret.height+1 : rret.height+1;
    return ret;
}

bool isBalanced(struct TreeNode* root){
   Ret ret = _isBalanced(root);
    return ret.isbalanced;
    }