跟着专注于计算机视觉的AndyJ的妈妈我学算法之每日一题leetcode111二叉树的最小深度

Tags tree | depth-first-search | breadth-first-search
这个题，二叉树的最小深度，也是BFS的经典题，是道容易题，要把情况考虑清楚。
这个题可以用BFS，也可以用DFS。都用来做一下，挺不错。
好了，
题目：

给定一个二叉树，找出其最小深度。

最小深度是从根节点到最近叶子节点的最短路径上的节点数量。

说明: 叶子节点是指没有子节点的节点。

示例:

给定二叉树 [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7
返回它的最小深度  2.

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/minimum-depth-of-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

直接写解法，记得多回头看。
代码，BFS：

class Solution:
    def minDepth(self, root: TreeNode) -> int:
        # bfs
        if not root: return 0
        queue = []
        queue.append(root)
        depth = 0
        while queue:
            depth += 1
            n = len(queue)
            for i in range(n):
                root = queue.pop(0)
                if not root.left and not root.right:
                    return depth
                if root.left:
                    queue.append(root.left)
                if root.right:
                    queue.append(root.right)

代码，DFS：

class Solution:
    def minDepth(self, root: TreeNode) -> int:
        # dfs
        if not root: return 0
        def dfs(root):
            if not root.left and not root.right:
                return 1
            
            depth_left = 0
            depth_right = 0
            if root.left:
                depth_left = dfs(root.left)
            if root.right:
                depth_right = dfs(root.right)
            if not depth_left:
                return depth_right + 1
            if not depth_right:
                return depth_left + 1
            return min(depth_left, depth_right)+1
        depth = dfs(root)
        return depth

好了，记得回头看，代码不难。温故知新。就这样吧。
不要怂，就是干。
未来是属于年轻人的。