【算法-LeetCode】498. 对角线遍历（二维数组）

498. 对角线遍历 - 力扣（LeetCode）

发布：2021年10月10日14:12:42

问题描述及示例

给你一个大小为 m x n 的矩阵 mat ，请以对角线遍历的顺序，用一个数组返回这个矩阵中的所有元素。

示例 1：

输入：mat = [[1,2,3],[4,5,6],[7,8,9]]
输出：[1,2,4,7,5,3,6,8,9]

示例 2：
输入：mat = [[1,2],[3,4]]
输出：[1,2,3,4]

提示：
m == mat.length
n == mat[i].length
1 <= m, n <= 104
1 <= m * n <= 104
-105 <= mat[i][j] <= 105

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/diagonal-traverse
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

我的题解

主要思路比较简单，就是在用双层 for 循环遍历 mat 矩阵时，让辅助函数 getOneDiagonal(mat, i, j) 获取 mat 数组中以坐标 [i, j] 为起点且方向为由左下至右上的一条对角线 diag，当然，这个起点统一选择了下面图示中的浅绿色单元格部分。

if 判断的作用

然后同时通过一个 count 变量来判断是否需要将当前所得对角线 diag 进行翻转操作。再将 diag 中的元素逐个放入 res 中。最后将 res 返回即可。

有关详解请看下方注释：

/**
 * @param {number[][]} mat
 * @return {number[]}
 */
var findDiagonalOrder = function (mat) {
  // res用于存储最终结果并作为最终返回值返回
  let res = [];
  // count用于记录当前是第几次获取对角线，可以控制是否要将获取到的对角线进行翻转操作
  let count = 0;
  // 开始遍历mat矩阵的第一列
  for (let i = 0; i < mat.length; i++) {
    // 获取以 [i, 0] 为起点的对角线
    let diag = getOneDiagonal(mat, i, 0);
    // 根据count的奇偶性来判断是否要将当前获得的对角线进行翻转操作后再翻入res中
    // 注意这里用到了扩展运算符
    count % 2 ? res.push(...diag.reverse()) : res.push(...diag);
    count++;
  }
  // 开始遍历mat的最后一行，注意 i 是从1开始的
  for (let i = 1; i < mat[0].length; i++) {
    // 下面的逻辑和上面的一样
    let diag = getOneDiagonal(mat, mat.length - 1, i);
    count % 2 ? res.push(...diag.reverse()) : res.push(...diag);
    count++;
  }
  // 最后返回res作为最终结果
  return res;

  // getOneDiagonal是一个辅助函数，用于获取arr数组中以 [i,j] 为起点一条对角线
  function getOneDiagonal(arr, i, j) {
    let diagonal = [];
    // 如果当前元素在矩阵上边界和右边界范围内
    while (i >= 0 && j < arr[0].length) {
      diagonal.push(arr[i][j]);
      // 注意该函数获取的对角线方向都是从左下至右上
      i--;
      j++;
    }
    // 将当前获取到的对角线最为返回值返回
    return diagonal;
  }
};


提交记录
32 / 32 个通过测试用例
状态：通过
执行用时：96 ms, 在所有 JavaScript 提交中击败了81.77%的用户
内存消耗：44.4 MB, 在所有 JavaScript 提交中击败了28.06%的用户
时间：2021/10/10 14:15

精简与优化

上面的程序中其实有许多逻辑都是重复的，只是有个别参数不同罢了，于是我就对上面的程序进行了一定程度的精简和优化。总体思路还是一样的。

/**
 * @param {number[][]} mat
 * @return {number[]}
 */
var findDiagonalOrder = function (mat) {
  let res = [];
  let count = 0;
  // 开始遍历 mat 矩阵元素
  for (let i = 0; i < mat.length; i++) {
    for(let j = 0; j < mat[0].length; j++) {
      // 判断当前元素是否为上面图示中的浅绿色单元格，（即矩阵的左边界和下边界）
      if((i < mat.length && j === 0) || (i === mat.length-1 && j < mat[0].length)) {
        // 如果是的话，则开始获取以当前元素为起点的对角线
        let diag = getOneDiagonal(mat, i, j);
        count % 2 ? res.push(...diag.reverse()) : res.push(...diag);
        count++;
      }
    }
  }
  return res;

  function getOneDiagonal(arr, i, j) {
    let diagonal = [];
    while (i >= 0 && j < arr[0].length) {
      diagonal.push(arr[i][j]);
      ++j && --i;
    }
    return diagonal;
  }
};


提交记录
32 / 32 个通过测试用例
状态：通过
执行用时：96 ms, 在所有 JavaScript 提交中击败了81.77%的用户
内存消耗：44.7 MB, 在所有 JavaScript 提交中击败了24.70%的用户
时间：2021/10/10 14:42

代码更精简了一点，但是似乎内存消耗更多了一点点。不过LeetCode的判题结果也不能太当真，有个大致参考就行了。

官方题解

更新：2021年7月29日18:43:21

因为我考虑到著作权归属问题，所以【官方题解】部分我不再粘贴具体的代码了，可到下方的链接中查看。

更新：2021年10月10日14:46:28

参考：对角线遍历 - 对角线遍历 - 力扣（LeetCode）

【更新结束】

有关参考

暂无