蓝桥杯python 回形取数

基础训练BASIC-25 回形取数

问题描述
　　回形取数就是沿矩阵的边取数，若当前方向上无数可取或已经取过，则左转90度。一开始位于矩阵左上角，方向向下。

输入格式
　　输入第一行是两个不超过200的正整数m, n，表示矩阵的行和列。接下来m行每行n个整数，表示这个矩阵。

输出格式
　　输出只有一行，共mn个数，为输入矩阵回形取数得到的结果。数之间用一个空格分隔，行末不要有多余的空格。

样例输入
3 3
1 2 3
4 5 6
7 8 9

样例输出
1 4 7 8 9 6 3 2 5

样例输入
3 2
1 2
3 4
5 6

样例输出
1 3 5 6 4 2

【我的思路(有错请指出~)】
以样例1为例：
1.首先取第一列，分别是arr[0][0],arr[1][0],arr[2][0]。arr[i].pop(0)即在arr中删去这个值，防止后面重复读取。
2.接下来取的是最后一行arr[-1]，arr.pop()为删去arr中的最后一个值，arr是一个二维数组，所以arr.pop()删去的是arr中的最后一个列表。
3.接下来取最后一列（从最后一行往上,倒序），取完即删除。
4.接下来取的是最新的第一行（从最后一列往前，倒序），取完即删除。
5.按照前四步循环取数并删除已取的数，直至arr为空（所有数都取完了）。
（如果不好理解，可以在草稿上画出二维数组，按照逻辑取数就明白了）

m,n = map(int,input().split())
arr = [[]*n]*m
list = []
for i in range(0,m):
    arr[i] = [int(j) for j in input().split()]

while arr:
    for i in range(0,len(arr)):
        list += [arr[i].pop(0)] if arr[i] else []
    list += arr.pop() if arr else  []
    for j in range(len(arr)-1,-1,-1):
        list += [arr[j].pop()] if arr[j] else []
    list += arr.pop(0)[::-1] if arr else []

print(*list)