B1050 螺旋矩阵（python）

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

AC
这道题要逻辑很清楚，通过序号的变换由外至内来给矩阵赋值~

import math
N = int(input())
if N == 1:
    print(input())
    exit()
if math.sqrt(N) % 1 == 0:
    m = int(math.sqrt(N))
else:
    m = int(math.sqrt(N) + 1)
while N % m != 0:
    m += 1
n = N // m
nums = [int(x) for x in input().split()]
nums.sort(reverse=True)# 将所有数字排序
matrix = [[0 for i in range(n)] for j in range(m)]# 构造mxn矩阵
cnt = 0
de, i, j = 0, 0, 0
while cnt < N:
    while cnt < N and i < n-1:# 从左到右添加
        matrix[j][i] = nums[cnt]
        i += 1
        cnt += 1
    while cnt < N and j < m-1:# 从上到下添加
        matrix[j][i] = nums[cnt]
        j += 1
        cnt += 1
    while cnt < N and i > de:# 从右到左添加
        matrix[j][i] = nums[cnt]
        i -= 1
        cnt += 1
    while cnt < N and j > de:# 从下到上添加
        matrix[j][i] = nums[cnt]
        j -= 1
        cnt += 1
    de, i, j, m, n = de+1, i+1, j+1, m-1, n-1
    if cnt == N-1:
        matrix[j][i] = nums[cnt]
        cnt += 1

for line in matrix:
    print(*line)

这道题一开始还想天真的排好序按行输出……不构造矩阵不行啊~还是看了一下大神的思路