24.平衡二叉树（简单）

最新推荐文章于 2022-07-31 10:17:23 发布

SCU-JJkinging

最新推荐文章于 2022-07-31 10:17:23 发布

阅读量181

点赞数 1

分类专栏：力扣刷题日志（简单）文章标签：二叉树数据结构算法 leetcode

本文链接：https://blog.csdn.net/qq_41845478/article/details/108976248

版权

力扣刷题日志（简单）专栏收录该内容

41 篇文章 2 订阅

订阅专栏

给定一个二叉树，判断它是否是高度平衡的二叉树。

本题中，一棵高度平衡二叉树定义为：

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

示例 1:

给定二叉树 [3,9,20,null,null,15,7]

返回 true 。

示例 2:

给定二叉树 [1,2,2,3,3,null,null,4,4]

返回 false 。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def isBalanced(self, root: TreeNode) -> bool:

        def height(node: TreeNode):
            '''定义一个求树的高度的函数'''
            if not node: # 若节点为空，则树高为0
                return 0
            else: # 否则返回左子树和右子树的最大高度，再加根节点的1
                return max(height(node.left), height(node.right)) + 1
        
        if not root or (root.left == None and root.right == None):
            return True # 若根节点为空或者根节点不空但是左右子树为空，则是平衡二叉树
        if abs(height(root.left) - height(root.right)) <= 1 and \
        self.isBalanced(root.left) and self.isBalanced(root.right):
            # 如果根节点的左右两个子树的高度差的绝对值不超过1, 并且左右子树也都是二叉平衡树
            # 则该树是平衡的
            return True
        else: return False # 否则不平衡