剑指offer系列-面试题-面试题55 - II. 平衡二叉树(python)

1. 题目

输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。

2. 解题思路

详情见 面试题55 - II. 平衡二叉树（从底至顶、从顶至底，清晰图解）

2.0 递归

整棵树是平衡二叉树，则每一个子树都是平衡二叉树，很明显可以使用递归。

3. 代码实现

3.0 递归

# 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:
        """
        递归，递归判断两棵子树是否为平衡二叉树
        """
        def recur(root):
            if not root: return 0, True # 返回二叉树的深度
            l, l_bal = recur(root.left) # 左子树的深度
            r, r_bal = recur(root.right) # 右子树的深度

            depth = l+1 if l > r else r+1 # 树的深度
            bal = l-r <= 1 if l > r else r-l <= 1
            return depth, bal and l_bal and r_bal # 当前树，左子树，右子树是否均平衡

        _, flag = recur(root)
        return flag

4. 总结

对于二叉树而言，递归比较好些，难的是迭代的咋写，俺不会呀、

5. 参考文献

[1] 剑指offer丛书
[2] 剑指Offer——名企面试官精讲典型编程题
[3] 面试题55 - II. 平衡二叉树（从底至顶、从顶至底，清晰图解）