一起刷力扣之【542. 01 矩阵】

这是我参与2022首次更文挑战的第28天，活动详情查看：2022首次更文挑战

题目

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

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

示例

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

输入： 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 <= 10$^4$
• 1 <= m * n <= 10$^4$
• mat[i][j] is either 0 or 1.
• mat 中至少有一个 0 

解题思路

这道题与昨天讲解的【1162. 地图分析】思路类似，可以说是 【1162. 地图分析】 的基础版。

1. 从左上到右下遍历，计算当前非 0 的节点与左上方最近的一个 0 的距离
2. 从右下到左上遍历，计算当前非 0 的节点与右下方最近的一个 0 的距离
3. 根据 步骤1，步骤2 计算出的结果，取最小值即可得到最终的答案

代码实现

class Solution {
    public int[][] updateMatrix(int[][] mat) {
        int m = mat.length, n = mat[0].length;
        int[][] result = new int[m][n];
        
        // 从左上到右下遍历
        for(int i = 0; i < m; ++i){
            for(int j = 0; j < n; ++j){
                if(mat[i][j] == 0){
                    // 当前位置为0，则最短距离为0
                    result[i][j] = 0;
                }else{
                    // 当前位置不为零，由于是第一次访问该节点，需要先初始化，便于后续比较
                    result[i][j] = m * n;
                    // 不是边界，则取前一距离来进行判断，取最小值
                    if(i > 0){
                        result[i][j] = Math.min(result[i][j], result[i - 1][j] + 1);
                    }
                    if(j > 0){
                        result[i][j] = Math.min(result[i][j], result[i][j - 1] + 1);
                    }
                }
            }
        }

        // 从右下到左上遍历
        for(int i = m - 1; i >= 0; --i){
            for(int j = n - 1; j >= 0; --j){
                // 去掉不需要判断的节点
                if(mat[i][j] != 0){
                    // 获取当前节点与右下方 0 的距离，将其与左上方 0 的距离做比较，取最小值
                    if(i < m - 1){
                        result[i][j] = Math.min(result[i][j], result[i + 1][j] + 1);
                    }
                    if(j < n - 1){
                        result[i][j] = Math.min(result[i][j], result[i][j + 1] + 1);
                    }
                }
            }
        }

        return result;
    }
}