力扣（LeetCode）- 542. 01 矩阵

题目：

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

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

示例 1：

|---|---|---|
| 0 | 0 | 0 |
| 0 | 1 | 0 |
| 1 | 1 | 1 |

复制代码

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

解答：javascript

主要思路就是，新建一个相同的数组 res，先将 0 位置填入，然后遍历所有 0 ，它位置的上下左右设为1，然后遍历所有 1 ，它位置的上下左右设为2，依次类推

javascript 复制代码

var updateMatrix = function(mat) {
    const res = []
    const visited = []
    for (let i = 0;i < mat.length;i++) {
      const r = []
      const v = []
      for (let j = 0;j < mat[0].length;j++) {
        if (mat[i][j]) {
          r.push(null)
          v.push(0)
        } else {
          r.push(0)
          v.push(1)
        }
      }
      res.push(r)
      visited.push(v)
    }
    dp(0, res, visited)
    return res
};

const dp = (n, res, visited) => {
  for (let i = 0;i < res.length;i++) {
    for (let j = 0;j < res[0].length;j++) {
      if (res[i][j] === n-1) {
        add(i, j, res, visited, n)
      }
    }
  }
  if (check(res)) {
    return
  }
  dp(n + 1, res, visited)
}

// 最近的元素距离0位置加1
const add = (i, j, res, visited, n) => {
  // left
  if (j - 1 >= 0) {
    set(i, j-1, res, visited, n)
  }
  // right
  if (j + 1 <= res[0].length - 1) {
    set(i, j+1, res, visited, n)
  }
  // up
  if (i - 1 >= 0) {
    set(i-1, j, res, visited, n)
  }
  // down
  if (i + 1 <= res.length - 1) {
    set(i+1, j, res, visited, n)
  }
}

const set = (i, j, res, visited, n) => {
  if (!visited[i][j]) {
    res[i][j] = n
    visited[i][j] = 1
  }
}

const check = (res) => {
  let flag = true
  
  for (let i = 0;i < res.length;i++) {
    for (let j = 0;j < res[0].length;j++) {
      if (res[i][j] === null) {
        flag = false
      }
    }
  }
  return flag
}