題目描述
給定一個二叉樹,找出其最小深度。
最小深度是從根節點到最近葉子節點的最短路徑上的節點數量。
說明: 葉子節點是指沒有子節點的節點。
算法思路
採用遞歸。
第一步:考慮結束條件
節點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