【資料結構】01矩陣 01 Matrix
阿新 • • 發佈:2021-10-05
目錄
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