题目:
给定一个由 0 和 1 组成的矩阵 mat ,请输出一个大小相同的矩阵,其中每一个格子是 mat 中对应位置元素到最近的 0 的距离。
两个相邻元素间的距离为 1 。
示例一:
输入:mat = [[0,0,0],[0,1,0],[0,0,0]]
输出:[[0,0,0],[0,1,0],[0,0,0]]
示例二:
输入:mat = [[0,0,0],[0,1,0],[1,1,1]]
输出:[[0,0,0],[0,1,0],[1,2,1]]
分析:
由于这个问题与无权图的最近距离相关,因此可以考虑应用广度优先遍历搜索解决,该问题换句话来说就是求每个格子距离最近的0的距离,因此可以将所有的0当作初始节点添加到队列中,然后以值为0的节点作为起点做广度优先搜索,如果经过d步到达某个格子,那么该格子离最近的0的距离就是d。
创建一个大小与输入矩阵相同的二维数组dists,用来记录最终结果也就是每个格子离最近的0的距离,如果matrix[i][j]为0,那么这个格子力最近的0的距离自然是0,如果matrix[i][j]为1,则先用最大的整数值初始化dists[i][j],接下来搜索到对应的节点再更新它的值。
如果之前已经到达过坐标为(r,s)的格子,那么dists[r][s]的值一定不可能大于dist+1,因为这是用的广度优先搜索,二广度优先搜索能够保证从起始节点到达任意节点一定是沿着最短路径的,当第一次到达坐标为(r,s)的格子时记录到dists[r][s]的值一定是从值为0的格子到达该格子的最短距离,因此当再次到达(r,s)的格子时,dist[r][s]>dist+1的值为false,通过比较距离可以避免重复访问某个格子,该算法时间复杂度为O(mn).
代码:
package com.kuang;
import java.util.LinkedList;
import java.util.Queue;
public class UpdateMatrix {
public int[][] updateMatrix(int[][] mat) {
int rows = mat.length;
int cols = mat[0].length;
int[][] dists = new int[rows][cols];
Queue<int[]> queue = new LinkedList<>();
for (int i = 0; i < rows; i++) {
for (int j = 0; j < cols; j++) {
if (mat[i][j] == 0){
queue.add(new int[]{i,j});
dists[i][j] = 0;
}else {
dists[i][j] = Integer.MAX_VALUE;
}
}
}
int[][] dirs = {{-1,0},{1,0},{0,-1},{0,1}};
while (!queue.isEmpty()){
int[] pos = queue.remove();
int dist = dists[pos[0]][pos[1]];
for (int[] dir : dirs){
int r = pos[0] + dir[0];
int s = pos[1] + dir[1];
if (r>=0&&s>=0&&r<rows&&s<cols){
if (dists[r][s] > dist +1){
dists[r][s] = dist+1;
queue.add(new int[]{r,s});
}
}
}
}
return dists;
}
}