542. 01 矩阵
给定一个由 0 和 1 组成的矩阵 mat ，请输出一个大小相同的矩阵，其中每一个格子是 mat 中对应位置元素到最近的 0 的距离。
两个相邻元素间的距离为 1 。
示例 1：
输入：mat = [[0,0,0],[0,1,0],[0,0,0]] 输出：[[0,0,0],[0,1,0],[0,0,0]]
示例 2：
输入：mat = [[0,0,0],[0,1,0],[1,1,1]] 输出：[[0,0,0],[0,1,0],[1,2,1]]
提示：
m == mat.lengthn == mat[i].length1 <= m, n <= 1041 <= m * n <= 104mat[i][j] is either 0 or 1.mat 中至少有一个 0 原站题解
golang 解法, 执行用时: 56 ms, 内存消耗: 7.8 MB, 提交时间: 2021-07-22 14:45:33
// 542. 01 矩阵
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
}