1. 程式人生 > >【Leetcode_總結】111. 二叉樹的最小深度 - python

【Leetcode_總結】111. 二叉樹的最小深度 - python

Q:

給定一個二叉樹,找出其最小深度。

最小深度是從根節點到最近葉子節點的最短路徑上的節點數量。

說明: 葉子節點是指沒有子節點的節點。

示例:

給定二叉樹 [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回它的最小深度  2.

 


連結:https://leetcode-cn.com/problems/minimum-depth-of-binary-tree/description/

思路:看到這個題目就想起了最大深度 —》 【Leetcode_總結】104. 二叉樹的最大深度 - python

 當時試試將最大深度的max替換成min 就會出問題,最大那個問題中,遞迴的終止條件是某一葉子節點不存在,也就是左葉子或右葉子不存在時,就返回1,但是這個題目的話,當根節點的葉子節點不存在時,另外一個分支依舊是有效深度,因此我們需要考慮:

  • 當根節點為空時,返回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 minDepth(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        if not root:
            return 0
        if  not root.right:
            return self.minDepth(root.left) + 1
        if not root.left:
            return self.minDepth(root.right) + 1
        return min(self.minDepth(root.left) + 1, self.minDepth(root.right) + 1)