JavaScriptのバイナリ行列で最も近い0の距離

バイナリ行列は、0または1のみを含む配列の配列です。バイナリ行列を唯一の引数として受け取るJavaScript関数を作成する必要があります。

この関数は、同じ数の行と列を含む新しい行列を作成する必要があり、元の行列の各要素について、結果の行列には、元の行列の0からのその要素の最も近い距離が含まれている必要があります。

距離を計算するときは、斜めではなく水平または垂直に移動できることに注意する必要があります。また、マトリックスに少なくとも1つの0が含まれていることが保証されています。

例-

入力行列が-

const arr = [
   [0, 0, 0]
   [0, 1, 0]
   [1, 1, 1]
];

その場合、出力行列は-

const output = [
   [0, 0, 0]
   [0, 1, 0]
   [1, 2, 1]
];

例

このためのコードは-

const arr = [
   [0, 0, 0],
   [0, 1, 0],
   [1, 1, 1],
];
const findNearestDistance = (arr = []) => {
   let array = [];
   let res = arr.map((el, ind) => el.map((subEl, subInd) => {
      if (subEl === 0) {
         array.push([ind, subInd])
         return 0
      };
      return Number.MAX_SAFE_INTEGER;
   }));
   const updateAdjacent = (ind, subInd, min, array = []) => {
      if (ind < 0 || subInd < 0 || ind == arr.length || subInd == arr[0].length){
         return;
      };
      if (res[ind][subInd] < min + 2) return
         res[ind][subInd] = min + 1
         array.push([ind, subInd])
   };
   while (array.length) {
      let next = []
      for (let [ind, subInd] of array) {
         updateAdjacent(ind, subInd + 1, res[ind][subInd], next)
         updateAdjacent(ind, subInd - 1, res[ind][subInd], next)
         updateAdjacent(ind + 1, subInd, res[ind][subInd], next)
         updateAdjacent(ind - 1, subInd, res[ind][subInd], next)
      };
      array = next;
   }
   return res;
};
console.log(findNearestDistance(arr));

出力

そして、コンソールの出力は-

[ [ 0, 0, 0 ], [ 0, 1, 0 ], [ 1, 2, 1 ] ]