LeetCode 542. 01 矩阵

542. 01 矩阵

难度 中等

给定一个由 0 和 1 组成的矩阵 mat ，请输出一个大小相同的矩阵，其中每一个格子是 mat 中对应位置元素到最近的 0 的距离。

两个相邻元素间的距离为 1 。

示例 1：

输入：mat = [[0,0,0],[0,1,0],[0,0,0]]
输出：[[0,0,0],[0,1,0],[0,0,0]]

示例 2：

输入：mat = [[0,0,0],[0,1,0],[1,1,1]]
输出：[[0,0,0],[0,1,0],[1,2,1]]

提示：

• m == mat.length
• n == mat[i].length
• 1 <= m, n <= 104
• 1 <= m * n <= 104
• mat[i][j] is either 0 or 1.
• mat 中至少有一个 0

题解

​ 这道题看上去不是很难，一般都使用广搜去做，刚开始我也是这样，就是两层遍历，每个点都广搜，最后果然是超时，这可把我难倒了，怎么去优化这个。开始想通过打表的，发现好像行不通，这也附上超时的代码吧。

class Solution {
    int[] dx = {1, -1, 0, 0};
    int[] dy = {0, 0, 1, -1};
    public int[][] updateMatrix(int[][] mat) {
        int r = mat.length;//多少行
        int c = mat[0].length;//多少列
        int[][] ans = new int[r][c];//距离矩阵
        int[][] visit = new int[r][c];//遍历矩阵
        for(int i = 0; i < r; i++){
            for(int j = 0; j < c; j++){
                if(mat[i][j] != 0){//只扫描不为0的点
                    Queue<int []> queue = new LinkedList<int []>();//辅助队列
                    queue.offer(new int []{i, j, 0});//第一个参数是行，第二个是列，第三个是距离
                    visit[i][j] = 1;//标记位已经访问
                    while(!queue.isEmpty()){//队列不为空
                        int[] now = queue.poll();//出队
                        int x = now[0];//航坐标
                        int y = now[1];//列坐标
                        int level = now[2];//距离
                        if(mat[x][y] == 0){//遍历到第一个0
                            ans[i][j] = level;//更新距离
                            init(visit, r, c);//恢复访问数组
                            break;
                        }
                        for(int k = 0; k < 4; k++){//上下左右移动
                            int nx = x + dx[k];
                            int ny = y + dy[k];
                            int nlevel = level + 1;
                            if(nx >= 0 && nx < r && ny >= 0 && ny < c && visit[nx][ny] ==0){
                                queue.offer(new int []{nx, ny, nlevel});
                                visit[x][y] = 1;
                            }
                        }
                    }
                }
            }
        }
        return ans;
    }

    public void init(int [][] visit, int r, int c){
        for(int i = 0; i < r; i++){
            for(int j = 0; j < c; j++){
                visit[i][j] = 0;
            }
        }
    }
}

​ 确实没办法了，就去看了官方的解答，思路刚好相反的。官方解法从0开始广搜，这样不会出现从1开始广搜的问题，这个问题就是重复搜索，这样就解决超时的问题了。

class Solution {
    int[] dx = {1, -1, 0, 0};
    int[] dy = {0, 0, 1, -1};
    public int[][] updateMatrix(int[][] mat) {
        int r = mat.length;//多少行
        int c = mat[0].length;//多少列
        int[][] ans = new int[r][c];//距离矩阵
        int[][] visit = new int[r][c];//遍历矩阵
        Queue<int []> queue = new LinkedList<int []>();//辅助队列
        for(int i = 0; i < r; i++){
            for(int j = 0; j < c; j++){
                if(mat[i][j] == 0){//扫描为0的点
                    queue.offer(new int[]{i,j});//入队
                    visit[i][j] = 1;//标记位访问
                }
            }
        }

        while(!queue.isEmpty()){//出队
            int[] now = queue.poll();
            int x = now[0];
            int y = now[1];
            for(int i = 0; i < 4; i++){//上下左右访问
                int nx = x + dx[i];
                int ny = y + dy[i];
                if(nx >=0 && nx < r && ny >=0 && ny < c && visit[nx][ny]==0){//判断是否边界和是否访问过
                    ans[nx][ny] = ans[x][y] + 1;//更新标记数组
                    queue.offer(new int[]{nx, ny});//入队
                    visit[nx][ny] = 1;//标记访问
                }
            }
        }
        return ans;
    }
}