1050 螺旋矩阵 (25 分)

本题要求将给定的 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

 

#include <cstdio>
#include <algorithm>
#include <cmath>

using namespace std;

bool myCmp(int a, int b) {
    return a > b;
}

int N, arr[10005], m, n;
int matrix[10005][10005];

int main() {
    scanf("%d", &N);
    for (int i = 0; i < N; ++i) {
        scanf("%d", arr + i);
    }
    sort(arr, arr + N, myCmp);
    int ns = sqrt(N);
    for (n = ns; n > 0; n--) {
        if (N % n == 0) {
            m = N / n;
            break;
        }
    }
    int m1 = m, n1 = n;
    int sx = 0, sy = 0;
    int cnt = 0;
    while (m > 0 && n > 0) {
        for (int i = 0; i < n; i++) {
            matrix[sx][sy + i] = arr[cnt++];
        }
        if (cnt == N) {
            break;
        }
        for (int i = 0; i < m - 2; i++) {
            matrix[sx + i + 1][sy + n - 1] = arr[cnt++];
        }
        if (cnt == N) {
            break;
        }
        for (int i = 0; i < n; i++) {
            matrix[sx + m - 1][sy + n - 1 - i] = arr[cnt++];
        }
        if (cnt == N) {
            break;
        }
        for (int i = 0; i < m - 2; i++) {
            matrix[sx + m - 2 - i][sy] = arr[cnt++];
        }
        sx++;
        sy++;
        m -= 2;
        n -= 2;
    }
    for (int i = 0; i < m1; i++) {
        printf("%d", matrix[i][0]);
        for (int j = 1; j < n1; ++j) {
            printf(" %d", matrix[i][j]);
        }
        printf("\n");
    }
    return 0;
}