题目:给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明: 叶子节点是指没有子节点的节点。
思路:与求二叉树的高度思路相似,不过要注意递归条件如下。
- 当节点为空,return 0
- 当节点左节点为空,return 节点右子树的最小深度。
- 当节点右节点为空,return 节点左子树最小深度。
- 当节点左右节点都不为空时,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