算法—leetcode—111

题目

（111.）二叉树的最小深度

题目描述

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

说明

叶子节点是指没有子节点的节点。

示例

示例一

输入：root = [3,9,20,null,null,15,7]
输出：2

示例二

输入：root = [2,null,3,null,4,null,5,null,6]
输出：5

思路一

使用BFS， 明确起点和终点
起点：二叉树的根节点
终点：最靠近根节点的叶子页面

叶子节点就是两个子节点都是nil的节点

if node.Left == nil && node.Right==nil {
	// 叶子节点
}

代码

// minDepth
// 二叉树最小深度
// BFS（广度优先搜索） 
func minDepthBFS(root *TreeNode) int {
    if root == nil {
        return 0
    }
    // 使用切片定义一个存储树结构的队列
    queue := []*TreeNode{}
    // 记录当前状态,
    count := []int{}
    
    queue = append(queue, root)
    // root 本身就是一层，depth 初始化为 1
    count = append(count, 1)
    
    // 将当前队列中的所有节点向四周扩散
    for i := 0; i < len(queue); i++ {
        log.Println("队列长度", len(queue), i, count[i])
        // 当前节点
        node := queue[i]
        // 深度
        depth := count[i]
        // 判断是否达到终点
        // 如果是叶子节点，则直接返回深度
        if node.Left == nil && node.Right == nil {
            return depth
        }
        
        // 将相邻的节点加入到队列中
        // 校验左子数是否为空
        if node.Left != nil {
            queue = append(queue, node.Left)
            count = append(count, depth + 1)
        }
        // 校验右子数是否为空
        if node.Right != nil {
            queue = append(queue, node.Right)
            count = append(count, depth + 1)
        }
    }
    return 0
}

思路二

使用DFS，递归求最小的路径

代码

// minDepth
// 二叉树最小深度
// DFS（深度优先搜索）
func minDepthDFS(root *TreeNode) int {
    if root == nil {
        return 0
    }
    // 叶子节点 
    if root.Left == nil && root.Right == nil {
        return 1
    }
    // 初始化一个最小值
    minD := math.MaxInt32
    // 递归求左子数最小深度
    if root.Left != nil {
        minD = min(minDepthDFS(root.Left), minD)
    }
    // 递归求右子数最小深度
    if root.Right != nil {
        minD = min(minDepthDFS(root.Right), minD)
    }
    return minD + 1
}

func min(x, y int) int {
    if x < y {
        return x
    }
    return y
}

参考

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

labuladong的算法小抄