Solution With JS

Author: bola-nabil

Medium/542. 01 Matrix.js

@@ -0,0 +1,23 @@
+var updateMatrix = function (mat) {
+	const m = mat.length,
+		n = mat[0].length;
+
+	for (let i = 0; i < m; i++) {
+		for (let j = 0; j < n; j++) {
+			if (mat[i][j] === 0) continue;
+			mat[i][j] = Infinity;
+			if (i - 1 >= 0) mat[i][j] = Math.min(mat[i][j], 1 + mat[i - 1][j]);
+			if (j - 1 >= 0) mat[i][j] = Math.min(mat[i][j], 1 + mat[i][j - 1]);
+		}
+	}
+
+	for (let i = m - 1; i >= 0; i--) {
+		for (let j = n - 1; j >= 0; j--) {
+			if (mat[i][j] === 0) continue;
+			if (i + 1 < m) mat[i][j] = Math.min(mat[i][j], 1 + mat[i + 1][j]);
+			if (j + 1 < n) mat[i][j] = Math.min(mat[i][j], 1 + mat[i][j + 1]);
+		}
+	}
+
+	return mat;
+};