题目
给你一个大小为 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]
思路
模拟题。可以看出来从第一行开始开始（这里的行指的是对角线），奇数次行从该行对角线左下到右上依次遍历，偶数次行从该对角线右上到左下依次遍历，下标的遍历规律很好找，奇数次（i-1, j+1），偶数次(i+1, j-1)。
重要的是边界坐标如何改变，可以看出上半部分奇数行到偶数行改变时，坐标向右移一位；偶数行到奇数行改变是，坐标向下移一位。但是下班部分正好相反，如何找到这个边界是关键，这里使用的是判断边界坐标向右移或者向下移是否越界，没有越界就按上半边规律走，越界了就按相反的规律走
题解
# 对角线输出 def findDiagonalOrder(mat: List[List[int]]) -> List[int]: index, ans, row = (0, 0), [], 1 while row <= len(mat) + len(mat[0]) - 1: # 奇数行 if row % 2 != 0: for i in range(row): ans.append(mat[index[0]][index[1]]) if 0 <= index[0] - 1 < len(mat) and 0 <= index[1] + 1 < len(mat[0]): index = (index[0]-1, index[1] +1) else: break # 边界坐标转换 if index[1] + 1 == len(mat[0]): index = (index[0] + 1, index[1]) else: index = (index[0], index[1] + 1) row += 1 else: # 偶数行 for i in range(row): ans.append(mat[index[0]][index[1]]) if 0 <= index[0] + 1 < len(mat) and 0 <= index[1] - 1 < len(mat[0]): index = (index[0] + 1, index[1] - 1) else: break # 边界坐标转换 if index[0] + 1 == len(mat): index = (index[0], index[1] + 1) else: index = (index[0] + 1, index[1]) row += 1 return ans
