找规律的题
def findDiagonalOrder(self, mat):
"""
:type mat: List[List[int]]
:rtype: List[int]
"""
#注意不一定是方阵
#先确定对角线的个数为m+n-1
#再确定每条对角线的走向,奇数往左下走,偶数忘右上走
#在确定每条对角线的起点,根据中线划分
#如果是奇数线,x的起点在对角线下标小于列的长度时为0,大于列的长度时就是对角线下标减去列的长度+1
#y的起点在对角线下标小于列的长度时为i,大于列的长度时就是最后一列
#如果是偶数线,x的起点在对角线下标小于行的长度时为i,大于行的长度时就是最后一行
#y的起点在对角线下标小于行的长度时为0,大于列的长度时就是对角线下标减去行的长度+1
res = []
m = len(mat)
n = len(mat[0])
for i in range(m+n-1):
if i %2 == 1:
x = 0 if i < n else i - n + 1
y = i if i < n else n -1
while x < m and y >= 0:
res.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 x >= 0 and y < n:
res.append(mat[x][y])
x -= 1
y += 1
return res