（python 3）1050 螺旋矩阵 (25分)

1050 螺旋矩阵 (25分)

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

输入格式：

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

输出格式：

输出螺旋矩阵。每行 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

 这个题最后一个测试点数值很大，一会全部通过一会超时很奇怪。可能我的思路使用C++实现会好一些叭。

大佬的不同思路：https://blog.csdn.net/A_D_E/article/details/92009183

我的python代码：

import math
import numpy as np


n=int(input())
a = [int(x) for x in input().split()]
a.sort(reverse=True)


#
n_shape_1=int(math.sqrt(n))
while n%n_shape_1 !=0:
    n_shape_1=n_shape_1-1
n_shape_0=n//n_shape_1


up=0
down=n_shape_0
left=0
right=n_shape_1


result=np.empty([n_shape_0,n_shape_1]).astype(int)


flag=0

#螺旋填充矩阵
#Spiral filled matrix
while True:
    for i in range(left,right):
        result[up][i]=a[flag]
        flag=flag+1
    up=up+1
    for j in range(up,down):
        result[j][right-1]=a[flag]
        flag=flag+1
    right=right-1
    for m in range(right-1,left-1,-1):
        result[down-1][m]=a[flag]
        flag=flag+1
    down=down-1
    if not flag<n:
        break
    for k in range(down-1,up-1,-1):
        result[k][left]=a[flag]
        flag=flag+1
    left=left+1
    

for p in result.tolist():
    print(*p)

薛定谔的测试点7（其实是很逼近时间上限了）：