【力扣日記】111 二叉樹的最小深度 | 遞歸

題目描述

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

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

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

算法思路

採用遞歸。

第一步:考慮結束條件
節點root爲空時結束,返回節點數 0
或節點爲葉子節點時,即root.left or root.right爲0

第二步:縮減規模
顯然是root.left 與root.right替代root遞歸。

第三步:返回值
返回值有兩種情況,
第一種左右節點都存在的情況下,此時節點要返回最小深度+1
第二種即某一節點爲空的情況,不特殊區分時會返回0,但顯然僅某一節點爲空並不是葉節點,不滿足條件,所以應該返回較大的那個節點值+1;又考慮到某一節點爲空,必然存在0,所以可以返回left+right+1.

舉例解釋一下第三步 第二種情況:對[1,2]這種樹來說最小深度顯然是2,但如果不區分情況,返回值一定是1的右節點+1=1

完整算法

class Solution:
    def minDepth(self, root: TreeNode) -> int:
        if not root:return 0
        if not(root.left or root.right):return 1
        left=self.minDepth(root.left)
        right=self.minDepth(root.right)
        if root.left and  root.right:return min(left,right)+1
        return left+right+1
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章