python(leetcode)498. 对角线遍历

这题难度中等，记录下思路

第一个会超时，

第二个：思想是按斜对角线行进行右下左上交替遍历，

 def traverse(matrix):
     n=len(matrix)-1
     m=len(matrix[0])-1
     result=[]
     for i in range(m+n+1):
             if(i % 2 == 0):
                 for j in range(i, -1, -1):
                     x=j
                     y=i-x
                     if x <= n and y <= m:
                         result.append(matrix[x][y])
                     # elif y > m:
                     #     break
                     else:
                         continue
             else:
                 for j in range(i, -1, -1):
                     y=j
                     x=i-y
                     if x <= n and y <= m:
                         result.append(matrix[x][y])
                     # elif x > n:
                     #     break
                     else:
                         continue
     return result

 if __name__ == '__main__':
     ma=[
  [ 1, 2, 3 ],
  [ 4, 5, 6 ],
  [ 7, 8, 9 ]
 ]
 print(traverse(ma))

11

 def diagonial(matrix):
     m=len(matrix)
     n=len(matrix[0])
     #思想是按斜对角线行进行遍历
     #遍历的方向右下，左上交替进行
     line=0
     res=[]
     tem=[]
     is_sure=True
     #对每一行进行循环 斜对角线有m+n-1条
     while line<(m+n-1):
     #设置起点
         if(line<m):
             x=line
             y=0
         else:
             x=m-1
             y=line-m-1
         #每行循环几次
         #因为都是从右上开始，所以设置条件x>0,y<n
         while x>=0 and y <n:
             # 存储斜行的元素
             tem.append(matrix[x][y])
             x-=1
             y+=1
         #如果是左下开始，则反向添加
         if(is_sure):
             tem.reverse()
         for num in tem:
             res.append(num)
         line+=1
     return res

 if __name__ == '__main__':
     matrix=[
  [ 1, 2, 3 ],
  [ 4, 5, 6 ],
  [ 7, 8, 9 ]
 ]
     print(diagonial(matrix))