题解 | #顺时针打印矩阵#
顺时针打印矩阵
http://www.nowcoder.com/practice/9b4c81a02cd34f76be2659fa0d54342a
借助一个辅助矩阵lab来记录元素是否被遍历,或者是否到了边界,然后按照题目规则去遍历整个矩阵,并保存到result中。
class Solution:
def printMatrix(self, matrix: List[List[int]]) -> List[int]:
# 判断matrix不为空
if len(matrix[0]) != 0:
# 构建matrix中每个元素的标签矩阵lab,表示该元素是否被遍历,初始值为True
# lab比matrix首尾多出两行两列,并初始化为False,便于判断坐标移动
lab = [[False for _ in range(len(matrix[0]) + 1)]]
for i in matrix:
x = [False]
for j in matrix[0]:
x.append(True)
x.append(False)
lab.append(x)
lab.append([False for _ in range(len(matrix[0]) + 1)])
# 构建输出用的list,命名为result
result = []
# towhere表示坐标换行移动方向,0为向左向右向下移动,1为向上移动,初始值为0。
towhere = 0
# 初始化坐标位置,x为行标,y为列标
x = 1
y = 1
# 开始遍历整个matrix
for i in range(len(matrix) * len(matrix[0])):
# 将当前坐标位置的数保存到reslut
result.append(matrix[x - 1][y - 1])
# 保存完了将lab中对应位置标签改为False,表示已遍历
lab[x][y] = False
# 开始移动坐标,先判断方向
if towhere == 0:
# 当前是向右下走,先考虑能不能向右走,若向右走,该位置没有被遍历则y+1
if lab[x][y + 1]:
y += 1
# 不能向右走,考虑向下走
else:
# 判断向下走,该位置有没有被遍历,没被遍历x+1
if lab[x + 1][y]:
x += 1
# 不能向下也不能向右走,考虑往左走
else:
# 判断向左走,该位置没有被遍历,则y-1
if lab[x][y - 1]:
y -= 1
# 不能向左走,则向上走
else:
# 判断向上走,该位置没有被遍历x-1
if lab[x - 1][y]:
x -= 1
# 此时将towhere改为1,因为此时改向上走了
towhere = 1
# 若此时向上走
else:
# 判断向上走,该位置没有被遍历x-1
if lab[x - 1][y]:
x -= 1
# 否则向右走,并将towhere改为0,因为接下去走到头就要向下换行
else:
y += 1
towhere = 0
return result
# 当matrix为空,直接返回空list
else:
return []