多源BFS,将矩阵中为0的元素存入队列,作为BFS的第一层,然后每遍历一层,就将当前层中的大小+1,需要创建一个辅助数组来记录每个结点是否有访问过。
class Solution {
public int[][] updateMatrix(int[][] matrix) {
int m = matrix.length;
int n = matrix[0].length;
Queue<int[]> q = new LinkedList<>();
boolean[][] visited = new boolean[m][n];
for(int i=0; i<m;i++){
for(int j =0; j<n;j++){
if(matrix[i][j] == 0){
q.offer(new int[]{matrix[i][j], i, j});
visited[i][j] = true;
}
}
}
int[][] dir = {{0,1},{0,-1},{1,0},{-1,0}};
int count = 0;
while(!q.isEmpty()){
int size = q.size();
for(int i=0; i<size;i++){
int[] cur = q.poll();
for(int j=0;j<4;j++){
int x = cur[1]+dir[j][0];
int y = cur[2]+dir[j][1];
if(x>=0&&x<m&&y>=0&&y<n&&!visited[x][y]){
matrix[x][y] = cur[0]+1;
visited[x][y] = true;
q.offer(new int[]{matrix[x][y], x, y});
}
}
}
}
return matrix;
}
}