平衡二叉树

ccornell

于 2023-07-13 14:45:50 发布

阅读量188

点赞数

分类专栏： LeetCode 文章标签：算法 leetcode

本文链接：https://blog.csdn.net/Bzdjsmz/article/details/131701701

版权

LeetCode 专栏收录该内容

97 篇文章 0 订阅

订阅专栏

该文介绍了一种判断二叉树是否为平衡二叉树的算法，通过计算各节点的左右子树深度，利用递归方法进行判断。当左右子树深度差不超过1时，树是平衡的。代码实现中定义了一个Solution类，包含isBalanced和depth两个方法，分别用于判断平衡和计算树的深度。

摘要由CSDN通过智能技术生成

题目描述

输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。

示例 1:
给定二叉树 [3,9,20,null,null,15,7]
在这里插入图片描述

返回 true 。

示例 2:

给定二叉树 [1,2,2,3,3,null,null,4,4]
在这里插入图片描述

返回 false 。

算法思想

对二叉树分情况讨论：
一、无左右子树，只有一个顶点结点，则此时为平衡二叉树。
二、此时有左右子树，求出左子树和右子树的高度，进行判断即可。

代码实现

class Solution {
public:
    bool isBalanced(TreeNode* root) { 
       if(!root) return true;
       int left = depth(root->left);
       int right = depth(root->right);
       if(abs(right - left) <= 1  && isBalanced(root->left) && isBalanced(root->right))
       {
           return true;
       }
       else
       {
           return false;
       }
    }
    int depth(TreeNode* root)
    {
        if(!root)
            return 0;
        int left_depth = depth(root->left);
        int right_depth = depth(root->right);
        return max(left_depth,right_depth) + 1;
    }
};