LeetCode498. 对角线遍历

LeetCode刷题记录

---

📜题目描述

给你一个大小为 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 <= 10^4
1 <= m * n <= 10^4
-10^5 <= mat[i][j] <= 10^5

💡解题思路

我们规定，在路径中有向上走和向下走
褐色的路线是向上走(左图），黄色的路线是向下走（右图）
向上走包括（红色字体）：
1、向上走到头，向右走：行坐标为0，并且列坐标不为n-1
2、向右走到头，向下走：列坐标为n-1
3、同时向右和向下：其他情况就是同时向右和向上

向下走包括（绿色字体）：
1、向左走到头，向下走：列坐标为0，并且行坐标不为n-1
2、向下走到头，向上走：行坐标为n-1
3、同时向左和向下：其余情况就是向左和向下

我们发现：
坐标和为偶数时 向上走
坐标和为奇数时，向下走
所以，从[0][0]开始填入数组，并判断下一次往哪走。
如果坐标和为偶数，就向上走。
如果坐标和为奇数，就向下走。然后根据上面所列的具体的情况再进行判断向右or向左or向上or向下

⌨C++代码