劍指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. 平衡二叉樹（從底至頂、從頂至底，清晰圖解）