542. 01 矩阵

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

---

题目描述

给定一个由 0 和 1 组成的矩阵，找出每个元素到最近的 0 的距离。

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

示例 1：

输入：
[[0,0,0],
[0,1,0],
[0,0,0]]

输出：
[[0,0,0],
[0,1,0],
[0,0,0]]
示例 2：

输入：
[[0,0,0],
[0,1,0],
[1,1,1]]

输出：
[[0,0,0],
[0,1,0],
[1,2,1]]

提示：

给定矩阵的元素个数不超过 10000。
给定矩阵中至少有一个元素是 0。
矩阵中的元素只在四个方向上相邻: 上、下、左、右。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/01-matrix
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

---

解题过程

解题思路

动态规划：由于是到0的距离，有可能是四个方向上到0的距离。因此利用动态规划，其中一个方向的状态转移应该是：
dp[i][j] = min(dp[i-1][j], dp[i][j-1]) + 1，mat[i][j]==1；
dp[i][j] = 0，mat[i][j] == 0;
另外三个方向的类似。综上所述，先从左上到右下更新，再从右下往左上更新，最后就能得到所求结果。

class Solution {
    public int[][] updateMatrix(int[][] mat) {
        //dp[i][j]表示(i,j)所对应的元素到最近0的距离
        int m = mat.length;
        int n = mat[0].length;
        int dp[][] = new int[m][n];

        //初始化
        for(int i = 0; i < m; i++){
            Arrays.fill(dp[i], Integer.MAX_VALUE/2);
        }

        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                if(mat[i][j] == 0){
                    dp[i][j] = 0;
                }else{
                    
                    if(i > 0){
                        dp[i][j] = Math.min(dp[i][j], dp[i - 1][j] + 1);
                    }
                    if(j > 0){
                        dp[i][j] = Math.min(dp[i][j], dp[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){
                    //dp[i][j] = 0;
                }else{
                    
                    if(i < m - 1){
                        dp[i][j] = Math.min(dp[i][j], dp[i + 1][j] + 1);
                    }
                    if(j < n - 1){
                        dp[i][j] = Math.min(dp[i][j], dp[i][j + 1] + 1);
                    }
                }
            }
        }

        return dp;

    }
}

总结