给定一个二叉树,找出其最小深度。
最小深度是从根节点到最近叶子节点的最短路径上的节点数量。
说明: 叶子节点是指没有子节点的节点。
思路:
采用递归实现,有以下几种情况:
1.根节点为空,深度为0
2.只有一个根节点。深度为1
3.左右子树皆不空,则返回1+左右子树中最小的深度。
4.左子树为空,则返回1+右子树深度。这里可能有点难以理解,可以想象成此时只有根节点a,以及其右子树b,此时最小深度为2。
5.右子树为空,则返回1+左子树深度。同上分析。
解答:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def minDepth(self, root: TreeNode) -> int:
if not root:
return 0
if root.left is None and root.right is None:
return 1
elif root.left is None:
return 1 + self.minDepth(root.right)
elif root.right is None:
return 1 + self.minDepth(root.left)
else:
left = self.minDepth(root.left)+1
right = self.minDepth(root.right)+1
return min(left, right)