【刷题】542.01矩阵（01 Matrix）

题目

给定一个由 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

来源：力扣（LeetCode）https://leetcode-cn.com/problems/01-matrix

思路

看到这个题后，第一个想法是用动态规划，还很华丽的写出了状态转移方程
$dp[i][j]=matrix[i][j]==0?0:min(dp[i+1][j],dp[i-1][j],dp[i][j+1],dp[i][j-1])+1$
原理很简单，如果matrix[i][j]为0，则说明该点最小距离为0，否则是其四周最小距离+1。
满心欢喜的写完了状态转移方程，但是咋实现？卡住了，灰溜溜的跑去看题解，发现我把问题写混了，应该是从左上到右下遍历一遍，再从右下到左上遍历一遍，这样就把一个位置的四个方向都考虑到了，也避免出现不知道如何比较的问题。

public int[][] updateMatrix(int[][] matrix) {
    int m = matrix.length;
    int n = matrix[0].length;
    int[][] dp = new int[m][n];
    for (int[] dpr : dp) {
        Arrays.fill(dpr, 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;
}

知识点

在做动态规划时，需要单向推动，不可能多个方向一齐推动这样没法进行结果累加了，多个方向时需要分开出来。