题目:
给定一个包含 m x n 个元素的矩阵(m 行, n 列),请按照顺时针螺旋顺序,返回矩阵中的所有元素。
示例 1:
输入: [ [ 1, 2, 3 ], [ 4, 5, 6 ], [ 7, 8, 9 ] ] 输出: [1,2,3,6,9,8,7,4,5]
示例 2:
输入: [ [1, 2, 3, 4], [5, 6, 7, 8], [9,10,11,12] ] 输出: [1,2,3,4,8,12,11,10,9,5,6,7]
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/spiral-matrix/
解法一:
遍历状态分为四种:
1.顶层 2.尾列 3.底层 4.首列
将遍历过得元素删除避免重复遍历,同时注意只剩一列时遍历尾列和首列的特殊性。
class Solution:
def spiralOrder(self, matrix: List[List[int]]) -> List[int]:
S = 0
lists = []
while(matrix):
if S == 0:
lists = lists+matrix[0]
del matrix[0]
S=1
elif S == 1:
h = len(matrix)
w = len(matrix[0])
if(w!=1):
for i in range(h):
lists.append(matrix[i][-1])
del matrix[i][-1]
else:
for i in range(h):
lists.append(matrix[0][0])
del matrix[0]
S = 2
elif S == 2:
lists = lists+matrix[-1][::-1]
del matrix[-1]
S=3
else:
h = len(matrix)
w = len(matrix[0])
if(w!=1):
for i in range(h-1, -1, -1):
lists.append(matrix[i][0])
del matrix[i][0]
else:
for i in range(h-1, -1, -1):
lists.append(matrix[i][0])
del matrix[i]
S = 0
return lists