4月打卡活动第15天 LeetCode第542题：01矩阵（中等）

4月打卡活动第15天 LeetCode第542题：01矩阵（中等）

• 题目：给定一个由 0 和 1 组成的矩阵，找出每个元素到最近的 0 的距离。
  两个相邻元素间的距离为 1 。
  

• 解题思路：先寻找十字方向上的距离，再寻找可通过其他节点的较短的距离。

class Solution {
    public int[][] updateMatrix(int[][] matrix) {
        int len1 = matrix.length;
        int len2 = matrix[0].length;
        int[][] ans = new int[len1][len2];
        for(int i=0;i<len1;i++){
            for(int j=0;j<len2;j++){
                int k=1;
                while(true){
                    if(matrix[i][j]==0){
                        ans[i][j] = 0;
                        break;
                    }
                    if((i-k>=0 && matrix[i-k][j]==0)||
                    (i+k<len1 && matrix[i+k][j]==0)||
                    (j-k>=0 && matrix[i][j-k]==0)||
                    (j+k<len2 && matrix[i][j+k]==0)){
                        ans[i][j] = k;
                        break;
                    }else{
                        k++;
                    }
                }
            }
        }
        for(int i=0;i<len1;i++){
            for(int j=0;j<len2;j++){
                if(ans[i][j]>1){
                    int min = 0;
                    if(i-1>=0){
                        min = ans[i-1][j];
                    }
                    if(j-1>=0){
                        if(min==0){
                            min = ans[i][j-1];
                        }else{
                            min = min<ans[i][j-1]?min:ans[i][j-1];
                        } 
                    }
                    if(j+1<len2){
                        if(min==0){
                            min = ans[i][j+1];
                        }else{
                            min = min<ans[i][j+1]?min:ans[i][j+1];
                        } 
                    }
                    if(i+1<len1){
                        if(min==0){
                            min = ans[i+1][j];
                        }else{
                            min = min<ans[i+1][j]?min:ans[i+1][j];
                        } 
                    }
                    ans[i][j] = Math.min((min+1),ans[i][j]);
                }
            }
        }
        return ans;
    }
}

• 题解做法：进行两次动态规划。

public int[][] updateMatrix(int[][] matrix) {
    if (matrix == null || matrix.length == 0) return null;
    int m = matrix.length, n = matrix[0].length;
    int[][] dp = new int[m][n];
    //初始化
    for (int i = 0; i < m; i++) {
        Arrays.fill(dp[i], 10001);
    }
    for (int i = 0; i < m; i++) {
        for (int j = 0; j < n; j++) {
            if (matrix[i][j] == 0) dp[i][j] = 0;
        }
    }
    //状态转移
    //第一次填表从左上到右下
    for (int i = 0; i < m; i++) {
        for (int j = 0; j < n; j++) {
            if (i - 1 >= 0) {
                dp[i][j] = Math.min(dp[i][j], dp[i - 1][j] + 1);
            }
            if (j - 1 >= 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 (i + 1 < m) {
                dp[i][j] = Math.min(dp[i][j], dp[i + 1][j] + 1);
            }
            if (j + 1 < n) {
                dp[i][j] = Math.min(dp[i][j], dp[i][j + 1] + 1);
            }
        }
    }
    return dp;
}

作者：jerrymouse1998
链接：https://leetcode-cn.com/problems/01-matrix/solution/javacong-yan-sou-dao-dpxiang-xi-zhu-jie-by-jerrymo/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。