LeetCode每日一题——498. 对角线遍历

题目

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

示例 1：

输入：mat = [[1,2,3],[4,5,6],[7,8,9]]
输出：[1,2,4,7,5,3,6,8,9]

示例 2：

输入：mat = [[1,2],[3,4]]
输出：[1,2,3,4]

思路

模拟题。可以看出来从第一行开始开始（这里的行指的是对角线），奇数次行从该行对角线左下到右上依次遍历，偶数次行从该对角线右上到左下依次遍历，下标的遍历规律很好找，奇数次（i-1, j+1），偶数次(i+1, j-1)。

重要的是边界坐标如何改变，可以看出上半部分奇数行到偶数行改变时，坐标向右移一位；偶数行到奇数行改变是，坐标向下移一位。但是下班部分正好相反，如何找到这个边界是关键，这里使用的是判断边界坐标向右移或者向下移是否越界，没有越界就按上半边规律走，越界了就按相反的规律走

题解

# 对角线输出
def findDiagonalOrder(mat: List[List[int]]) -> List[int]:
    index, ans, row = (0, 0), [], 1
    while row <= len(mat) + len(mat[0]) - 1:
    # 奇数行
        if row % 2 != 0:
            for i in range(row):
                ans.append(mat[index[0]][index[1]])
                if 0 <= index[0] - 1 < len(mat) and 0 <= index[1] + 1 < len(mat[0]):
                    index = (index[0]-1, index[1] +1)
                else:
                    break
            # 边界坐标转换
            if index[1] + 1 == len(mat[0]):
                index = (index[0] + 1, index[1])
            else:
                index = (index[0], index[1] + 1)
            row += 1
        else:
        # 偶数行
            for i in range(row):
                ans.append(mat[index[0]][index[1]])
                if 0 <= index[0] + 1 < len(mat) and 0 <= index[1] - 1 < len(mat[0]):
                    index = (index[0] + 1, index[1] - 1)
                else:
                    break
            #  边界坐标转换
            if index[0] + 1 == len(mat):
                index = (index[0], index[1] + 1)
            else:
                index = (index[0] + 1, index[1])
            row += 1

    return ans