Rayman的绝顶之路——Leetcode每日一题打卡1

Leetcode542. 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

注意:

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

想法思路：

1. 首先得到一个0，1二维数组，创建一个与该二维数组一样大小的二维数组并初始化，初始化时对该二维数组原本为0的地方设为0，原本有数字的地方设为一个最大值（上限+1）
2. 然后进行第一次遍历，从左到右、从上到下重新赋值；
3. 然后进行第二次遍历，从右到左、从下到上重新赋值。

具体代码实现如下所示：

import java.util.Arrays;

public class Leetcode542_01 {
    static class Solution {
        int[][] updateMatrix(int[][] matrix) {
            if (matrix == null || matrix.length == 0) return null;
            int m = matrix.length, n = matrix[0].length;
            int[][] res = new int[m][n];
            //初始化
            for (int i = 0; i < m; i++) {
                Arrays.fill(res[i], 10001);
            }
            for (int i = 0; i < m; i++) {
                for (int j = 0; j < n; j++) {
                    if (matrix[i][j] == 0) res[i][j] = 0;
                }
            }
            //动态规划
            //从左到右、从上到下重新赋值
            for (int i = 0; i < m; i++) {
                for (int j = 0; j < n; j++) {
                    if (i - 1 >= 0) {
                        res[i][j] = Math.min(res[i][j], res[i - 1][j] + 1);
                    }
                    if (j - 1 >= 0) {
                        res[i][j] = Math.min(res[i][j], res[i][j - 1] + 1);
                    }
                }
            }
            //从右到左、从下到上重新赋值
            for (int i = m - 1; i >= 0; i--) {
                for (int j = n - 1; j >= 0; j--) {
                    if (i + 1 < m) {
                        res[i][j] = Math.min(res[i][j], res[i + 1][j] + 1);
                    }
                    if (j + 1 < n) {
                        res[i][j] = Math.min(res[i][j], res[i][j + 1] + 1);
                    }
                }
            }
            return res;
        }
    }

    public static void main(String[] args) {
        Solution solution = new Solution();
        int[][] a = {
                {0, 0, 0},
                {0, 1, 0},
                {1, 1, 1}};
        int[][] b = solution.updateMatrix(a);
        for (int i = 0; i < b.length; i++) {
            for (int j = 0; j < b[0].length; j++) {
                System.out.print(b[i][j] + " ");
            }
            System.out.println();
        }
    }
}

2020.4.15打卡