剑指 Offer II 107. 矩阵中的距离（中等 图 bfs 动态规划 数组 矩阵）

剑指 Offer II 107. 矩阵中的距离

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

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

示例 1：

输入：mat = [[0,0,0],[0,1,0],[0,0,0]]
输出：[[0,0,0],[0,1,0],[0,0,0]]
示例 2：

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

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

分析

1、求图的最短路径的问题一般采用广度优先搜索，要用到队列，队列的参数显然要用int[]，那么需要考虑图的哪些节点可以当作初始节点添加到队列中，需要知道，问题是求每个格子离最近的 0 的距离，可以将所有的0当作初始节点添加到队列中，然后以值为 0 的节点作为起始点做 bfs，如果经过 d 步到达某个格子，那么该格子离最近的 0 的距离就是 d
2、那么接下来需要考虑如何得到 d，一般来说我们需要一个二维数组标记访问过的节点，注意到该题的输出是个二维数组，d 是填入到二维数组中的，那么我们在用一个二维数组存储 d 的同时标记访问过的节点（矩阵中值为 1 的点）
3、代码实现思路。第一步要将初始节点放入队列 q 中，同时初始化储存答案的二维数组 ans，遍历矩阵，将值为 0 的点存入 q，值为 1 时，ans 填入Integer.MAX_VALUE；第二步进行 bfs，创建一个 dirs 数组用来对节点上下左右移动，找到矩阵中每个值为 1 的节点距离最近的 0 的距离 d 填入到 ans 中（这一步用到了动态规划的思想，在 bfs 的过程中会先更新离 0 最近的节点的 d，后续的邻接点会根据其附近的 d 选择最小值）

题解（Java）

class Solution {
   
    public int[][] updateMatrix(int[][] mat)