题目

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

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

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过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.求树的深度
2.判断某个节点是否平衡
3.判断整个树是否平衡

# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None

class Solution:
    def isBalanced(self, root):
        """ :type root: TreeNode :rtype: bool """
        def dfs(root):
            if root==None:
                return 0
            return 1 + max(dfs(root.left),dfs(root.right))

        def floor(head): #判断节点的深度
            if head == None:
                return 0
            return 1+max(dfs(head.left),dfs(head.right))

        def isOneBalance(head): #判断某个节点是否平衡
            print(head.val)
            print(floor(head.left),floor(head.right))
            if abs(floor(head.left)-floor(head.right))<=1:
                print('true')
                return True
            print('false')
            return False
        if root:
            if isOneBalance(root):
                if self.isBalanced(root.left) and self.isBalanced(root.right):
                    return True
                else:
                    return False
            else:
                return False
        else:
            return True        

学习他人代码

class Solution:
    def isBalanced(self, root):
        """ :type root: TreeNode :rtype: bool """
        return self.help(root) != -1;

    def help(self, root): #判断root左右子树相差多少，-1代表差值大于2,0代表无差值
        if not root:
            return 0

        len1 = self.help(root.left)
        if len1 == -1: #如果左子树不平衡，返回-1
            return -1

        len2 = self.help(root.right)
        if len2 == -1: #如果右子树不平衡，返回-1
            return -1

        if abs(len1 - len2) > 1: #如果当前节点不平衡，返回-1
            return -1

        return max(len1, len2) + 1 #如果当前节点是平衡的，就返回以当前节点为根二叉树的层数