1050 螺旋矩阵分数 25(python3)极其简单的思路

1050 螺旋矩阵

分数 25

作者 CHEN, Yue

单位 浙江大学

本题要求将给定的 N 个正整数按非递增的顺序，填入“螺旋矩阵”。所谓“螺旋矩阵”，是指从左上角第 1 个格子开始，按顺时针螺旋方向填充。要求矩阵的规模为 m 行 n 列，满足条件：m×n 等于 N；m≥n；且 m−n 取所有可能值中的最小值。

输入格式：

输入在第 1 行中给出一个正整数 N，第 2 行给出 N 个待填充的正整数。所有数字不超过 104，相邻数字以空格分隔。

输出格式：

输出螺旋矩阵。每行 n 个数字，共 m 行。相邻数字以 1 个空格分隔，行末不得有多余空格。

输入样例：

12
37 76 20 98 76 42 53 95 60 81 58 93

输出样例：

98 95 93
42 37 81
53 20 76
58 60 76

代码长度限制

16 KB

时间限制

200 ms

内存限制

64 MB

此题的解点最重要的是观察元素的插入顺序，通过观察就可以得到是右→下→左→上插入，第一次将外围的填满以后，我们就需要进入内围，此时就需要按照右、下、左、上的顺序，判断目标方向上是否为0，否则换方向，直到找到有0的方向填满位置。因此外层循环需要i达到n-1即说明数据都已插入。

n=int(input())
num=[int(i) for i in input().split()]
#降序排序(从大到小)
num.sort(reverse=True)
#求行、列数
m=int(n**0.5)
while n%m!=0:
    m+=1
r=m
c=n//m
#若行数小于列数则交换(题目要求：行数≥列数)
if r<c:
    r,c=c,r
#初始化列表，将所有位置都赋0
res=[[0]*c for _ in range(r)]
#i用于控制题给列表赋值，x、y用于定位当前的位置。
i=0
x=0
y=0
res[x][y]=num[i]
#我们可以看出来，数据是从大到小依次是右→下→左→上，只需要循环寻找满足条件的赋值即可。
#y+1是向右移动，x+1是向下，y-1是向左，x-1是向上。我们需要保证不超出界限并且需要判断
#目标方向上是否有0。
while i<n-1:
    while(y+1<c and res[x][y+1]==0):y+=1;i+=1;res[x][y]=num[i]
    while(x+1<r and  res[x+1][y]==0):x+=1;i+=1;res[x][y]=num[i]
    while(y-1>=0 and res[x][y-1]==0):y-=1;i+=1;res[x][y]=num[i]
    while(x-1>=0 and res[x-1][y]==0):x-=1;i+=1;res[x][y]=num[i]
#*号用于解包列表或元组中的元素。它会将可迭代对象（如列表或元组）中的元素拆分为单独的参数。
#此处用于解包每行元素，简化代码。
for i in range(r):
    print(*res[i])