声明:只求用最简单的方法通过,不求优化,不喜勿喷
1.用dp来做
2.初始化一个dp数组,对于arr的每一个值,从上到先从左往右遍历,然后arr[i][j]是0的话,dp赋值为0,否则,dp赋值为左边和上边的最小值+1
3.再从下往上,从右往左遍历一次,如果右边和下边的值+1小于第一次遍历已经赋予的值,那么更新它
4.因为第一次遍历的时候右边和下边还没有值,所以需要两次遍历
5.最后输出数组即可
6.优化todo
/**
* @param {number[][]} mat
* @return {number[][]}
*/
var updateMatrix = function (mat) {
let dp = [];
for (var i = 0; i < mat.length; i++) {
dp.push([]);
}
for (var i = 0; i < mat.length; i++) {
for (var j = 0; j < mat[i].length; j++) {
if (mat[i][j] == 0) {
dp[i][j] = 0;
} else {
let min = 999999;
if (i - 1 >= 0 && dp[i - 1][j] < min) {
min = dp[i - 1][j]
}
if (j - 1 >= 0 && dp[i][j - 1] < min) {
min = dp[i][j - 1]
}
dp[i][j] = min + 1;
}
}
}
for (var i = mat.length - 1; i >= 0; i--) {
for (var j = mat[i].length - 1; j >= 0; j--) {
if (mat[i][j] == 0) {
dp[i][j] = 0;
} else {
if (i + 1 < mat.length && dp[i][j] > dp[i + 1][j] + 1) {
dp[i][j] = dp[i + 1][j] + 1
}
if (j + 1 < mat[i].length && dp[i][j] > dp[i][j + 1] + 1) {
dp[i][j] = dp[i][j + 1] + 1
}
}
}
}
return dp;
};