110.平衡二叉树

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

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

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

解法1 递归

常规

类似前序遍历
每一次递归调用时 考察

1. 当前节点 的左右子树高度差是否不超过1
2. 当前节点的左子树 的左右子树高度差是否不超过1
3. 当前节点的右子树 的左右子树高度差是否不超过1

# 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: TreeNode) -> bool:
        
        if not root:
            return True
        
        # 判断 当前节点 的左右子树是否高度平衡
        left_height = self.height(root.left)
        right_height = self.height(root.right)
        if abs(left_height - right_height) > 1:
            return False
        
        # 判断 当前节点的左子树 的左右子树是否高度平衡
        left = self.isBalanced(root.left)
        # 判断 当前节点的右子树 的左右子树是否高度平衡
        right = self.isBalanced(root.right)
        
        # 三者都平衡 以当前节点为根节点的二叉树 才是高度平衡的
        return left and right

        
    def height(self, node):
        
        if not node:
            return 0
        
        left = self.height(node.left) + 1
        right = self.height(node.right) + 1
        
        return max(left, right)
        

trick

求某一节点的树的高度，当该节点的树不是平衡二叉树时，令该树的高度为-1
使得递归一轮height就得到结果
常规写法需要递归多轮

# 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: TreeNode) -> bool:
        
        return self.height(root) != -1

    
    def height(self, node):
    
        if not node:
            return 0

        left = self.height(node.left) + 1
        right = self.height(node.right) + 1
        
        # 以当前节点为根节点的树 不是平衡二叉树
        if left == -1 or right == -1 or abs(left - right) > 1:
            return -1

        return max(left, right)
        

解法2 迭代

…