劍指offer系列-面試題-面試題55 - I. 二叉樹的深度(python)

1. 題目

輸入一棵二叉樹的根節點，求該樹的深度。從根節點到葉節點依次經過的節點（含根、葉節點）形成樹的一條路徑，最長路徑的長度爲樹的深度。

2. 解題思路

詳情見 面試題55 - I. 二叉樹的深度（後序遍歷、層序遍歷，清晰圖解）

2.1 遞歸

從根節點來看，整個樹的深度=max(左子樹的深度，右子樹的深度) + 1，就這樣遞歸下去，遞歸的基線條件就是節點爲None。

2.2 廣度優先遍歷(BFS)

二叉樹有幾層深度就是幾，所以每有一層深度就加1.

3. 代碼實現

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 maxDepth(self, root: TreeNode) -> int:
        l = self.maxDepth(root.left)
        r = self.maxDepth(root.right)

        return l+1 if l > r else r+1

3.2 BFS

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

class Solution:
    def maxDepth(self, root: TreeNode) -> int:
        """
        什麼是二叉樹的深度？
        每有一層，深度就加1
        """
        if not root: return 0

        depth = 0
        queue = [root]

        while queue:
            temp = []
            for node in queue:
                if node.left: temp.append(node.left)
                if node.right: temp.append(node.right)
            
            queue = temp
            depth += 1
        return depth
        

4. 總結

通常樹的問題都可以用遞歸來解決。

5. 參考文獻

[1] 劍指offer叢書
[2] 劍指Offer——名企面試官精講典型編程題