在JavaScript矩阵中对角线遍历
问题:
我们需要编写一个JavaScript函数,该函数采用一个方矩阵(具有相同行数和列数的数组的数组)。函数应该对角地遍历该数组数组,并准备一个新的元素数组,并按照在遍历时遇到的顺序进行放置。
例如,如果函数的输入为-
const arr = [ [1, 2, 3], [4, 5, 6], [7, 8, 9] ];
那么输出应该是-
const output = [1, 2, 4, 7, 5, 3, 6, 8, 9];
示例
为此的代码将是-
const arr = [ [1, 2, 3], [4, 5, 6], [7, 8, 9] ]; const findDiagonalOrder = (arr = []) => { if(!arr.length){ return []; }; let ind = 0; let colBegin = 0, rowBegin = 0; let rowMax = arr.length, colMax = arr[0].length; const res = [], stack = []; while(rowBegin< rowMax || colBegin=0 ; row++,col--){ if(ind%2 === 0){ stack.push((arr[row][col])); }else{ res.push(arr[row][col]); }; }; ind++; while(stack.length){ res.push(stack.pop()); }; colBegin++ if(colBegin> colMax-1 && rowBegin < rowMax){ colBegin = colMax-1 rowBegin++ } }; return res }; console.log(findDiagonalOrder(arr));
代码说明:
我们采取的步骤是-
在一个方向上移动,跟踪起点。
如果index是偶数,我们将压入堆栈并在到达对角线末端时弹出,然后将popd添加到我们的输出数组中。
当我们移到下一个对角线时,我们保持索引递增。
我们递增列的开始索引,直到到达终点为止,因为它在下一次迭代中将在最后一个索引处停止,并且我们将从此点开始递增行开始的索引。
输出结果
控制台中的输出将是-
[ 1, 2, 4, 7, 5, 3, 6, 8, 9 ]