python顺时针螺旋矩阵_（Python 3）1050螺旋矩阵（25点）,python31050,分

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++实现会好一些叭。

我的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

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(其实是很逼近时间上限了)：