LeetCode-111- 二叉树的最小深度(python)

二叉树的最小深度

题目:给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明: 叶子节点是指没有子节点的节点。

思路:与求二叉树的高度思路相似,不过要注意递归条件如下。

  1. 当节点为空,return 0
  2. 当节点左节点为空,return 节点右子树的最小深度。
  3. 当节点右节点为空,return 节点左子树最小深度。
  4. 当节点左右节点都不为空时,rerturn 左右子树最小深度。
    说明:注意当节点有一个节点为空而另一个节点不为空,如2,3条时,不应该返回左右子树的最小深度,因为最小深度的定义是从根节点到最近叶子节点的最短路径上的节点数量,如果在2,3这两种情况返回左右子树的最小深度就会导致得到的是非叶子节点的最小深度不满足题意。
    代码如下:
class Solution:
    def minDepth(self, root: TreeNode) -> int:
        if root is None or root.val is None:
            return 0
        if root.left is None and root.right is None:
            return 1
        if root.left is None:
            return self.minDepth(root.right) + 1
        if root.right is None:
            return self.minDepth(root.left) + 1
        return min(self.minDepth(root.left), self.minDepth(root.right)) + 1
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容