代码随想录 -- 二叉树 -- 平衡二叉树

110. 平衡二叉树 - 力扣（LeetCode）

思路：仍然是递归调用

1. 定义一个递归函数 count 用来计算二叉树的层数

2. isBalanced 函数：如果传入根节点为空返回真；如果根节点 | 左子树的层数 - 右子树的层数 | 大于1，返回假；最后返回根节点左子树、右子树是否是平衡二叉树。

class Solution(object):
    def count(self,root):
        if root==None:
            return 0
        else:
            return 1+max(self.count(root.left),self.count(root.right))

    def isBalanced(self, root):
        if root==None:
            return True
        if abs(self.count(root.left) - self.count(root.right))>1:
            return False
        return self.isBalanced(root.left) and self.isBalanced(root.right)