54. 螺旋矩阵

Tony Wey

于 2024-08-07 10:27:07 发布

阅读量64

点赞数 4

分类专栏： LeetCode刷题文章标签： python leetcode

本文链接：https://blog.csdn.net/qq_38023194/article/details/140979587

版权

LeetCode刷题专栏收录该内容

50 篇文章 0 订阅

订阅专栏

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]

解题思路

可以模拟螺旋矩阵的路径。初始位置是矩阵的左上角，初始方向是向右，当路径超出界限或者进入之前访问过的位置时，顺时针旋转，进入下一个方向。

判断路径是否进入之前访问过的位置需要使用一个与输入矩阵大小相同的辅助矩阵 visited，其中的每个元素表示该位置是否被访问过。当一个元素被访问时，将 visited 中的对应位置的元素设为已访问。

如何判断路径是否结束？由于矩阵中的每个元素都被访问一次，因此路径的长度即为矩阵中的元素数量，当路径的长度达到矩阵中的元素数量时即为完整路径，将该路径返回。

def spiralOrder(matrix):
	if not matrix or not matrix[0]:
		return
		
	m, n = len(matrix), len(matrix[0])
	visited = [[False] * n for i in range(m)]
	total = m * n
	order = [0] * total
	
	directions = [[1,0], [0,1], [-1,0], [0,-1]]
	row, col, dir_inx = 0, 0, 0
	
	for i in range(total):
		order[i] = matrix[row][col]
		visited[row][col] = True
		next_row, next_col = directions[dir_inx][0], directions[dir_inx][1]
		if not (0<=next_row<m and 0<=next_col<n and not visited[next_row][next_col]):
			dir_inx = (dir_inx + 1) % 4
		row, col = row + directions[dir_inx][0], col + directions[dir_inx][1]
	
	return order