【数组】498. 对角线遍历

【题目】

给你一个大小为 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]

【题解】

题解1：

• 思路

       看注释

• 复杂度

         时间复杂度：O(n)，空间复杂度：O(1)

• 代码

class Solution:
    def findDiagonalOrder(self, mat: List[List[int]]) -> List[int]:
        ans = []
        # 矩阵m，n
        m, n = len(mat), len(mat[0])
        # 第n条对角线
        for i in range(m + n - 1):
            # 当n为偶数时从下往上
            if i % 2:
                # 起始点
                x = 0 if i < n else i - n + 1
                y = i if i < n else n - 1
                while x < m and y >= 0:
                    ans.append(mat[x][y])
                    x += 1
                    y -= 1
            # 当n为奇数时从上往下
            else:
                # 起始点
                x = i if i < m else m - 1
                y = 0 if i < m else i - m + 1
                while x >= 0 and y < n:
                    ans.append(mat[x][y])
                    x -= 1
                    y += 1
        return ans