01矩阵 01 Matrix
在一个由 0和1 组成的矩阵mat,输出一个大小相同的矩阵，其中每个格子是mat中对应位置元素到最近的0的距离。
两个相邻元素间的距离是1.
mat = [[0,0,0],[0,1,0],[0,0,0]]
out：[[0,0,0],[0,1,0],[0,0,0]]
思路
mat是一个m*n阶矩阵，要求输出相同的矩阵，矩阵的每一个元素，代表mat中对应元素距离0元素的最近距离。
使用BFS,将mat所有的0元素坐标信息全部入队。并且使用一个visit二维数组记录结果，-1代表未访问过，0代表访问过并且距离最小是0.
在进行BFS过程中，每出队一个坐标，就对该坐标进行方向扩展，即寻找附近的元素坐标。如果访问过，就标记visit数组的值，并将该坐标信息+visit数组的值入队。
public class Data{
public int i;
public int j;
public int val;
Data(int x, int y,int v){
i =x;
j =y;
val= v;
}
}
public int[][] updateMatrix(int[][] mat) {
int n = mat.length;
int m = mat[0].length;
Queue<Data> queue = new LinkedList<Data>();
int[][] visit = new int[n][m];
//queue
//Visit[][]
initqueue(mat,n,m,queue,visit);
int[][] dir = {{1,0},{-1,0},{0,-1},{0,1}};
while(!queue.isEmpty()){
Data t = queue.peek();
for(int k =0;k<4;k++){
int x = t.i+dir[k][0];
int y = t.j+dir[k][1];
if(x<0||y<0||x>n-1||y>m-1){
continue;//out of board
}
if(visit[x][y]!=-1){
continue;// this visited
}
visit[x][y] = t.val+1;
queue.offer(new Data(x,y,visit[x][y]));
}
queue.poll();
}
return visit;
}
public void initqueue(int[][] mat,int n,int m,Queue queue,int[][] visit){
for(int i =0;i<n;i++){
for(int j =0;j<m;j++){
if(mat[i][j]==0){
queue.offer(new Data(i,j,0));
visit[i][j]=0;
}
else{
visit[i][j]=-1;
}
}
}
}
Tag
BFS