Leetcode刷题Python110. 平衡二叉树

Posted 2022-08-18 Better Bench

1 题目

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

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

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

2 解析

高度的函数，即可判断二叉树是否平衡。具体做法类似于二叉树的前序遍历，即对于当前遍历到的节点，首先计算左右子树的高度，如果左右子树的高度差是否不超过 11，再分别递归地遍历左右子节点，并判断左子树和右子树是否平衡。这是一个自顶向下的递归的过程。

3 Python 实现

class Solution:
    def isBalanced(self, root: Optional[TreeNode]) -> bool:
        
        def height(root: TreeNode) -> int:
            if not root:
                return 0
            return max(height(root.left), height(root.right)) + 1
        if not root:
            return True
        if self.isBalanced(root.left) and self.isBalanced(root.right):
            if abs(height(root.left) - height(root.right))<=1:
                return True
            else:
                return False
        return False