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

diag1-grid.jpg

使用了笨方法，每次都是从右上到左下，奇数次反着加入数组。

class Solution:
    def findDiagonalOrder(self, mat: List[List[int]]) -> List[int]:

        m = len(mat)
        n = len(mat[0])

            # 1 2 3
            # 4 5 6
            # 7 8 9

        arr = []
        
        for i in range(n):
            x = 0
            y = i

            temp = []
            while y > -1 and x < m:

                # print('x, y:', x, y)
                # print('mat', mat[x][y])

                temp.append(mat[x][y])

                x += 1
                y -= 1

            if i % 2 != 0:
                for v in temp:
                    arr.append(v)
            else:
                for v in temp[::-1]:
                    arr.append(v)

        for i in range(1, m):
            x = i
            y = n-1
            
            temp = []
            while y > -1 and x < m:
                # print('x, y:', x, y)
                # print('mat', mat[x][y])

                temp.append(mat[x][y])

                x += 1
                y -= 1

            if (n+i) % 2 == 0:
                for v in temp:
                    arr.append(v)
            else:
                for v in temp[::-1]:
                    arr.append(v)                

        return arr