Leetcode 54 螺旋矩阵

给你一个 m 行 n 列的矩阵 matrix ，请按照 顺时针螺旋顺序 ，返回矩阵中的所有元素。

示例 1：

 

 

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

 

 

输入：matrix = [[1,2,3,4],[5,6,7,8],[9,10,11,12]]
输出：[1,2,3,4,8,12,11,10,9,5,6,7]
 

提示：

m == matrix.length
n == matrix[i].length
1 <= m, n <= 10
-100 <= matrix[i][j] <= 100

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/spiral-matrix
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

class Solution(object):
    def spiralOrder(self, matrix):
        """
        :type matrix: List[List[int]]
        :rtype: List[int]
        """

        n, m = len(matrix[0]), len(matrix)  #n为矩阵的列，m为矩阵的行
        dir = [(0, 1), (1, 0), (0, -1), (-1, 0)]    #因为该题是顺时针取数，所以这样定向量坐标
        d = 0                               #该变量是用来改变方向
        cnt = 1                             #该变量使用来判断是否已经取完了所有数
        x, y = 0, 0                         #初始坐标
        ls = []
        ans = [[0] * n for _ in range(m)]   #标记数组
        while cnt <= n * m:
            ls.append(matrix[x][y])
            cnt += 1
            ans[x][y] = -1                  #该店取过后，标记为-1
            nx, ny = x + dir[d][0], y + dir[d][1]
            if nx < 0 or nx >= m or ny < 0 or ny >= n or ans[nx][ny] == -1:   #判断是否要改变方向
                d = (d + 1) % 4            #改变方向
                x = x + dir[d][0]
                y = y + dir[d][1]
            else:
                x, y = nx, ny

        return ls