给定一个由 0 和 1 组成的矩阵,找出每个元素到最近的 0 的距离。
两个相邻元素间的距离为 1 。
示例 1:
输入:
0 0 0
0 1 0
0 0 0
输出:
0 0 0
0 1 0
0 0 0
示例 2:
输入:
0 0 0
0 1 0
1 1 1
输出:
0 0 0
0 1 0
1 2 1
注意:
给定矩阵的元素个数不超过 10000。
给定矩阵中至少有一个元素是 0。
矩阵中的元素只在四个方向上相邻: 上、下、左、右。
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/01-matrix
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
抄的作业,具体解释请参看官方题解
class Solution {
public int[][] updateMatrix(int[][] matrix) {
int m = matrix.length;
int n= matrix[0].length;
//使用队列来保存所有的0
Queue<int[]> queue = 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){
queue.offer(new int[]{i,j});
visited[i][j] = true;
}
}
}
//按上下左右四个方向搜索未被访问过的点
int[] dx = {-1,1,0,0};
int[] dy = {0,0,-1,1};
while(!queue.isEmpty()){
int[] tem = queue.poll();
int i = tem[0],j=tem[1];
for(int k=0;k<4;k++){
int x=i+dx[k];
int y=j+dy[k];
if(x>=0&&x<m&&y>=0&&y<n&&!visited[x][y]){
matrix[x][y]=matrix[i][j]+1;
visited[x][y]=true;//标记
queue.offer(new int[]{x,y});//放入到队列中
}
}
}
return matrix;
}
}