题目:Given a binary tree, find its minimum depth.
The minimum depth is the number of nodes along the shortest path from the root node down to the nearest leaf node.
题目分析,求最短的时候的记得分类讨论,对于左子树为None和右子树为None的各种情况进行讨论,否则有一边的子树为空的时候,就容易使根节点的最短路径为0,而与实际情况不符合。
python代码实现:
class Solution(object):
def minDepth(self, root):
"""
:type root: TreeNode
:rtype: int
"""
if root == None:
return 0
if root.left == None and root.right == None:
return 1
if root.left == None:
return 1 + self.minDepth(root.right)
if root.right == None:
return 1 + self.minDepth(root.left)
return 1 + min(self.minDepth(root.left), self.minDepth(root.right))
大佬的代码:
class Solution:
# @param root, a tree node
# @return an integer
def minDepth(self, root):
if root == None:
return 0
if root.left==None or root.right==None:
return self.minDepth(root.left)+self.minDepth(root.right)+1
return min(self.minDepth(root.right),self.minDepth(root.left))+1
大佬把我中间分类讨论的部分进行了重新表述,这样可以更加的简洁,因为0的值并不影响到全局的问题。可以把他加上,这样原来分类讨论的二项可以合二为一。
在有一个值为0的时候,可以考虑到这种写法。