// 多源广度优先搜索
// 重点是全部值为0的加入，然后遍历一步，大小为1的得到。
// 再次bfs，得到距离0距离为2
// 其实这里应该建立一个map防止重复bfs的
func updateMatrix(matrix [][]int) [][]int {

n, m := len(matrix), len(matrix[0])
queue := make([][]int, 0)
for i := 0; i < n; i++ {    // 把0全部存进队列，后面从队列中取出来，判断每个访问过的节点的上下左右，直到所有的节点都被访问过为止。
    for j := 0; j < m; j++ {
        if matrix[i][j] == 0 {
            point := []int{i, j}
            queue = append(queue, point)
        } else {
            matrix[i][j] = -1
        }
    }
}
direction := [][]int{{0, 1}, {0, -1}, {1, 0}, {-1, 0}}

for len(queue) > 0 {  // 这里就是 BFS 模板操作了。
    point := queue[0]
    queue = queue[1:]
    for _, v := range direction {
        x := point[0] + v[0]
        y := point[1] + v[1]
        if x >= 0 && x < n && y >= 0 && y < m && matrix[x][y] == -1 {
            matrix[x][y] = matrix[point[0]][point[1]] + 1
            queue = append(queue, []int{x, y})
        }
    }
}

return matrix

}

作者：youaresb
链接：https://leetcode-cn.com/problems/01-matrix/solution/01ju-zheng-bfs-by-youaresb/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。