Python描述 LeetCode 498. 对角线遍历 大家好，我是亓官劼（qí guān jié ），在【亓官劼】公众号、GitHub、B站等平台分享一些技术博文，主要包括前端开发、python后端开发、小程序开发
Python描述 LeetCode 498. 对角线遍历
大家好，我是亓官劼（qí guān jié ），在【亓官劼】公众号、GitHub、B站等平台分享一些技术博文，主要包括前端开发、python后端开发、小程序开发、数据结构与算法、docker、Linux常用运维、NLP等相关技术博文，时光荏苒，未来可期，加油~
如果喜欢博主的文章可以关注博主的个人公众号【亓官劼】(qí guān jié)，里面的文章更全更新更快。如果有需要找博主的话可以在公众号后台留言，我会尽快回复消息.
本文原创为【亓官劼】（qí guān jié ），请大家支持原创，部分平台一直在恶意盗取博主的文章！！！ 全部文章请关注微信公众号【亓官劼】。
题目
给你一个大小为 m x n 的矩阵 mat ，请以对角线遍历的顺序，用一个数组返回这个矩阵中的所有元素。
示例 1：
输出：[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的图模拟一下就好，这里m和n不一样最好，比较方便推。
Python描述
class Solution:def findDiagonalOrder(self, mat: List[List[int]]) -> List[int]:
n,m,i,j = len(mat),len(mat[0]),0,0
res = []
for cir in range(n+m-1):
if cir%2 == 0:
i,j =cir if cir < n else n-1 ,0 if cir < n else cir - n + 1
while i >= 0 and j < m:
res.append(mat[i][j])
i,j = i-1,j+1
else:
i,j =0 if cir < m else cir - m + 1 ,cir if cir < m else m-1
while j >= 0 and i < n:
res.append(mat[i][j])
i,j = i+1,j-1
return