498. 对角线遍历 中等

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

输入：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

方法一：直接模拟
思路与算法

分对角线为奇数和偶数的情况考虑，对角线总数为m+n-1，对角线为奇数的时候从上往下遍历，对角线是偶数的时候从下往上遍历，注意起点，if else 语句。一个一个遍历并且加入结果列表即可。注意从上往下遍历的时候，每次行数+1列数-1，从下往上遍历的时候每次行数-1列数+1，最后要求行数和列数不超过mn以及》=0【限制条件】。

class Solution:
    def findDiagonalOrder(self, mat: List[List[int]]) -> List[int]:
        m,n,result=len(mat),len(mat[0]),[]
        if len(mat)==0:
            return []
        for i in range(m+n-1):
            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:
                    result.append(mat[x][y])
                    x+=1
                    y-=1
            else:
                x=i if i<m else m-1
                y=0 if i<m else i-m+1
                while y<n and x>=0:
                    result.append(mat[x][y])
                    x-=1
                    y+=1
        return result