- 螺旋矩阵I:按螺旋顺序返回矩阵所有元素
可以模拟螺旋矩阵的路径。初始位置是矩阵的左上角,初始方向是向右,当路径超出界限或者进入之前访问过的位置时,顺时针旋转,进入下一个方向。
判断路径是否进入之前访问过的位置需要使用一个与输入矩阵大小相同的辅助矩阵 visited,其中的每个元素表示该位置是否被访问过。当一个元素被访问时,将visited 中的对应位置的元素设为已访问。
如何判断路径是否结束?由于矩阵中的每个元素都被访问一次,因此路径的长度即为矩阵中的元素数量,当路径的长度达到矩阵中的元素数量时即为完整路径,将该路径返回。
class Solution:
def spiralOrder(self, matrix: List[List[int]]) -> List[int]:
dirs = [(0, 1), (1, 0), (0, -1), (-1, 0)]
m = len(matrix)
n = len(matrix[0])
visited = [[0] * n for _ in range(m)]
res_list = []
row, col, dirIdx = 0, 0, 0
for i in range(m * n): # 矩阵中的每个元素都被访问一次
res_list.append(matrix[row][col])
visited[row][col] = 1
dx, dy = dirs[dirIdx]
r, c = row + dx, col + dy
if r < 0 or r >= m or c < 0 or c >= n or visited[r][c] == 1:
dirIdx = (dirIdx + 1) % 4 # 顺时针旋转至下一个方向
dx, dy = dirs[dirIdx]
row, col = row + dx, col + dy
return res_list
- 螺旋矩阵II:模拟矩阵的生成
按照要求,初始位置设为矩阵的左上角,初始方向设为向右。若下一步的位置超出矩阵边界,或者是之前访问过的位置,则顺时针旋转,进入下一个方向。如此反复直至填入n2个元素。
Q:如何设置已经访问的位置,避免再次访问?
记 matrix 为生成的矩阵,其初始元素设为 0。由于填入的元素均为正数,我们可以判断当前位置的元素值,若不为 0,则说明已经访问过此位置。
Q:如何顺时针旋转进入下一方向?
class Solution:
def generateMatrix(self, n: int) -> List[List[int]]:
dirs = [(0, 1), (1, 0), (0, -1), (-1, 0)]
matrix = [[0] * n for _ in range(n)]
row, col, dirIdx = 0, 0, 0
for i in range(n * n):
matrix[row][col] = i + 1
dx, dy = dirs[dirIdx]
r, c = row + dx, col + dy
if r < 0 or r >= n or c < 0 or c >= n or matrix[r][c] > 0:
dirIdx = (dirIdx + 1) % 4 # 顺时针旋转至下一个方向
dx, dy = dirs[dirIdx]
row, col = row + dx, col + dy
return matrix