leetcode 542. 01 矩阵【BFS】

多源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;
    }
}

你可能感兴趣的:(刷题路漫漫,leetcode,bfs,矩阵)