最后的判断可以优化一下,之前采用了三个if判断,不过也无所谓
class Solution(object):
def minDepth(self, root):
"""
:type root: TreeNode
:rtype: int
"""
if not root:
return 0
ldepth=self.minDepth(root.left)+1
rdepth=self.minDepth(root.right)+1
if ldepth==1 or rdepth==1:
return max(ldepth,rdepth)
return min(ldepth,rdepth)