JavaScript의 행렬에서 가장 긴 1 줄 찾기

다음과 같은 이진 행렬(0 또는 1만 포함하는 배열의 배열)이 있다고 가정합니다.

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

첫 번째이자 유일한 인수로 이러한 행렬을 취하는 JavaScript 함수를 작성해야 합니다.

우리 함수의 임무는 행렬에서 연속된 것 중 가장 긴 줄을 찾고 그 안에 있는 1의 개수를 반환하는 것입니다. 선은 수평, 수직, 대각선 또는 역대각선일 수 있습니다.

예를 들어, 위 배열의 경우 출력은 -

const output = 3

가장 긴 줄은 arr[0][1]에서 시작하여 대각선으로 -

arr[2][3]

예시

이에 대한 코드는 -

const arr = [
   [0,1,1,0],
   [0,1,1,0],
   [0,0,0,1]
];
const longestLine = (arr = []) => {
   if(!arr.length){
      return 0;
   }
   let rows = arr.length, cols = arr[0].length;
   let res = 0;
   const dp = Array(rows).fill([]);
   dp.forEach((el, ind) => {
      dp[ind] = Array(cols).fill([]);
      dp[ind].forEach((undefined, subInd) => {
         dp[ind][subInd] = Array(4).fill(null);
      });
   });
   for (let i = 0; i < rows; i++) {
      for (let j = 0; j < cols; j++) {
         if (arr[i][j] == 1) {
            dp[i][j][0] = j > 0 ? dp[i][j - 1][0] + 1 : 1;
            dp[i][j][1] = i > 0 ? dp[i - 1][j][1] + 1 : 1;
            dp[i][j][2] = (i > 0 && j > 0) ? dp[i - 1][j - 1][2] + 1 : 1;
            dp[i][j][3] = (i > 0 && j < cols - 1) ? dp[i - 1][j + 1][3] + 1 : 1;
            res = Math.max(res, Math.max(dp[i][j][0], dp[i][j][1]));
            res = Math.max(res, Math.max(dp[i][j][2], dp[i][j][3]));
         };
      };
   };
   return res;
};
console.log(longestLine(arr));

출력

콘솔의 출력은 -

3