蓝桥杯 基础练习 回形取数（Python）

资源限制

内存限制：512.0MB   C/C++时间限制：1.0s   Java时间限制：3.0s   Python时间限制：5.0s

问题描述

　　回形取数就是沿矩阵的边取数，若当前方向上无数可取或已经取过，则左转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

r, c = map(int, input().split())
a = []
for i in range(r):
    a.append(list(map(int, input().split())))
mu = []  # 存储取出的数
direction = 0  # 记录取数方向，（0,1,2,3）
while True:
    if len(a) == 0 or len(a[0]) == 0:  # 行或列为0时退出
        break
    if direction == 0:  # 方向0，取第1列，从上到下，并且删除
        for i in range(len(a)):
            mu.append(a[i].pop(0))
    elif direction == 1:  # 方向1，取最后一行，从左到右，并且删除
        for i in range(len(a[0])):
            mu.append(a[len(a) - 1][i])
        a.pop(len(a) - 1)
    elif direction == 2:  # 方向2，取最后列，从下到上，并且删除
        for i in range(len(a) - 1, -1, -1):
            mu.append(a[i].pop(len(a[i]) - 1))
    elif direction == 3:  # 方向3，取第1行，从右到左，并且删除
        for i in range(len(a[0]) - 1, -1, -1):
            mu.append(a[0][i])
        a.pop(0)
    direction = (direction + 1) % 4
for i in range(len(mu)):  # 按要求输出结果
    if len(mu) == 1:
        print(mu.pop(0))
    else:
        print(mu.pop(0), end=" ")