题目:原题链接(中等)

标签:数组

解法 时间复杂度 空间复杂度 执行用时
Ans 1 (Python) O ( M × N ) O(M×N) O(M×N) O ( 1 ) O(1) O(1) 208ms (76.56%)
Ans 2 (Python)
Ans 3 (Python)

解法一:

class Solution:
    def findDiagonalOrder(self, matrix: List[List[int]]) -> List[int]:
        if not matrix or not matrix[0]:
            return []

        m, n = len(matrix), len(matrix[0])
        i, j, d1, d2 = 0, 0, -1, 1

        ans = []

        while len(ans) < m * n:
            ans.append(matrix[i][j])
            if 0 <= i + d1 < m and 0 <= j + d2 < n:
                i, j = i + d1, j + d2
            else:
                if j + d2 >= n:
                    i += 1
                elif i + d1 < 0:
                    j += 1
                elif i + d1 >= m:
                    j += 1
                elif j + d2 < 0:
                    i += 1
                d1, d2 = d2, d1

        return ans