【力扣日记】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
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章