【树-简单】面试题 04.04. 检查平衡性

【题目】
实现一个函数，检查二叉树是否平衡。在这个问题中，平衡树的定义如下：任意一个节点，其两棵子树的高度差不超过 1。

示例 1:

给定二叉树 [3,9,20,null,null,15,7]
3
/
9 20
/
15 7
返回 true 。

示例 2:

给定二叉树 [1,2,2,3,3,null,null,4,4]
1
/
2 2
/
3 3
/
4 4
返回 false 。
【代码】
【方法1】

class Solution:
    def visit(self,root):
        if not root:
            return 0
        L=self.visit(root.left)
        R=self.visit(root.right)
        if abs(L-R)>1:
            self.flag=False
        return max(L,R)+1
    def isBalanced(self, root: TreeNode) -> bool:
        self.flag=True
        self.visit(root)
        return self.flag

【方法2】

class Solution:
    def visit(self,root):
        if not root:
            return 0
        L=self.visit(root.left)
        R=self.visit(root.right)
        if abs(L-R)>1 or L==-1 or R==-1:
            return -1
        return max(L,R)+1
    def isBalanced(self, root: TreeNode) -> bool:        
        return self.visit(root)>=0

【方法3】

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