leetcode 111. 二叉树的最小深度 (第一遍刷题)

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

解题思路：
分三种情况讨论

• 当左右子节点都有的时候
• 当只有左子树的时候
• 当只有右子树的时候
  注意：不能不分类讨论，而只用min()方法来求最小深度。

# Definition for a binary tree node.
class TreeNode(object):
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

class Solution(object):
    # 要分情况讨论
    def minDepth(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        if root:
            if ((root.left and root.right)):#当左右子节点都有的时候
                return 1 + min(self.minDepth(root.left), self.minDepth(root.right))
            elif root.left:#当只有左子树的时候，则深度加1并递归左子树
                return 1 + self.minDepth(root.left)
            elif root.right:#当只有右子树的时候，则深度加1并递归右子树
                return 1 + self.minDepth(root.right)
            else:
                #当这一层已经是最后一层的时候，深度加1，不进行递归操作
                return 1
        else:
            return 0

n1 = TreeNode(1)
n2 = TreeNode(2)

n1.left = n2
s = Solution()
result = s.minDepth(n1)
print(result)