leetcode498. 对角线遍历

找规律的题

 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