题意:给定一个m行,n列的矩阵,按螺旋顺序返回所有元素的值。
For example,
Given the following matrix:
[
[ 1, 2, 3 ],
[ 4, 5, 6 ],
[ 7, 8, 9 ]
]
You should return [1,2,3,6,9,8,7,4,5].
思路:碰到这类矩阵的题,还是要从索引的范围和规律入手,我一开始是想把矩阵转化为ndarry,就可以直接取一行的值了,之后按48题的思路做,但是后来仔细一看,原来这题不是方阵,哎,接看代码吧。
方法一:
def spiralOrder(self, matrix):
if not matrix or not matrix[0]:
return []
ans = []
m, n = len(matrix), len(matrix[0])
u, d, l, r = 0, m - 1, 0, n - 1
while l < r and u < d:
ans.extend([matrix[u][j] for j in xrange(l, r)])
ans.extend([matrix[i][r] for i in xrange(u, d)])
ans.extend([matrix[d][j] for j in xrange(r, l, -1)])
ans.extend([matrix[i][l] for i in xrange(d, u, -1)])
u, d, l, r = u + 1, d - 1, l + 1, r - 1
if l == r:
ans.extend([matrix[i][r] for i in xrange(u, d + 1)])
elif u == d:
ans.extend([matrix[u][j] for j in xrange(l, r + 1)])
return ans
方法二:
def spiralOrder(self, matrix):
return matrix and list(matrix.pop(0)) + self.spiralOrder(zip(*matrix)[::-1])
这方法简直了,但是貌似不能用于太大的矩阵,因为解包并不是很方便,但是这个递归的思路确实挺别致的。