蓝桥杯基础练习VIP-回形取数

题目描述

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

代码

m,n = map(int,input().split())
nums = []
for i in range(m):
    num = list(map(int,input().split()))
    nums.append(num)

get_nums = []
flag=1  # 1下，2右，3上，4左
cut_m = 0
cut_n = 0
i = -1
j = 0
while(len(get_nums)<m*n):
    if flag==1:
        for c in range(m-cut_m):
            i += 1
            get_nums.append(nums[i][j])
        cut_n += 1
        flag = 2
    elif flag==2:
        for c in range(n-cut_n):
            j += 1
            get_nums.append(nums[i][j])
        cut_m += 1
        flag = 3
    elif flag==3:
        for c in range(m-cut_m):
            i -= 1
            get_nums.append(nums[i][j])
        cut_n += 1
        flag = 4
    elif flag==4:
        for c in range(n-cut_n):
            j -= 1
            get_nums.append(nums[i][j])
        cut_m += 1
        flag = 1

for c in get_nums:
    print(c,end=' ')