给你一个 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