spiral_matrix.py

classSolution:
defspiralOrder(self, matrix: List[List[int]]) ->List[int]:

ifmatrix== []:
return []

test_matrix= [ [ 1, 2 ], [ 4, 5 ], [ 7, 8 ] ]
print( len(test_matrix) )
print( len(test_matrix[0]) )

top=0
bottom=len(matrix)-1
left=0
right=len(matrix[0])-1

new_list= []

# we define the directions
# 0: go right
# 1: go down
# 2: go left
# 3: go up
direction=0

whileTrue:
# 0: go right
ifdirection==0:
foriinrange(left, right+1):
new_list.append(matrix[top][i])
top+=1
# 1: go down
elifdirection==1:
foriinrange(top, bottom+1):
new_list.append(matrix[i][right])
right-=1
# 2: go left
# note: be careful about the 'range'
elifdirection==2:
foriinrange(right, left-1, -1):
new_list.append(matrix[bottom][i])
bottom-=1
# 3: go up
elifdirection==3:
foriinrange(bottom, top-1, -1):
new_list.append(matrix[i][left])
left+=1

# change the direction!!!
direction= (direction+1) %4

iftop>bottomorleft>right:
returnnew_list