题目:
给定一个由 0 和 1 组成的矩阵 mat ,请输出一个大小相同的矩阵,其中每一个格子是 mat 中对应位置元素到最近的 0 的距离。
两个相邻元素间的距离为 1
题解:
对于矩阵中的任意一个 1 以及一个 0,我们如何从这个 1 到达 0 并且距离最短呢?根据上面的做法,我们可以从 1 开始,先在水平方向移动,直到与 0 在同一列,随后再在竖直方向上移动,直到到达 0 的位置。这样一来,从一个固定的 1 走到任意一个 0,在距离最短的前提下可能有四种方法:
只有 水平向左移动 和 竖直向上移动;
只有 水平向左移动 和 竖直向下移动;
只有 水平向右移动 和 竖直向上移动;
只有 水平向右移动 和 竖直向下移动。
然后根据这四种情况作出四种方向的循环:
class Solution {
static int [][]dir={
{-1,0},{1,0},{0,-1},{0,1}};
public int[][] updateMatrix(int[][] mat) {
int m=mat.length,n=mat[0].length;//如果合理-1,那么下面的for循环那里就要<=