题目:
给定一个包含 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
就是螺旋状的遍历矩阵,每次到了边界就换方向。
注意边界值,每遍历完一行或一列要更新边界值。
代码:
class Solution(object):
def spiralOrder(self, matrix):
"""
:type matrix: List[List[int]]
:rtype: List[int]
"""
if not matrix:
return []
m=len(matrix)
n=len(matrix[0])
up,left=0,0
down,right=m-1,n-1
ans=[]
while len(ans)<=m*n:
for i in range(left,right+1):
ans.append(matrix[up][i])
up+=1#遍历完最上边一行,更新上边界
if up>down:break
for i in range(up,down+1):
ans.append(matrix[i][right])
right-=1#遍历完最右边一列,更新右边界
if right<left:break
i=right
while i>=left:
ans.append(matrix[down][i])
i-=1
down-=1#遍历完最下边一行,更新下边界
if down<up:break
i=down
while i>=up:
ans.append(matrix[i][left])
i-=1
left+=1#遍历完最左边一列,更新左边界
if left>right:break
return ans