PAT 1050 螺旋矩阵

https://pintia.cn/problem-sets/994805260223102976/problems/994805275146436608

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

输入格式：

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

输出格式：

输出螺旋矩阵。每行 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<bits/stdc++.h>
using namespace std;

int a[10000][1000], s[10000];

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

int main() {
    int n;
    scanf("%d", &n);
    for(int i = 0; i < n; i ++)
    scanf("%d", &s[i]);
    sort(s, s + n, cmp);

    int r, c;
    int minn = 99999;
    for(int i = 1; i <= sqrt(n * 1.0); i ++) {
        if(n % i == 0) {
            if(n / i - i < minn) {
                minn = n / i - i;
                r = i;
            }
        }
    }
    c = n / r;

    a[1][1] = s[0];
    int tot = 0;
	int x = 1, y = 1;
    while(tot < r * c - 1) {
        while(y + 1 <= r && ! a[x][y + 1])
            a[x][++ y] = s[++ tot];
        while(x + 1 <= c && !a[x + 1][y])
            a[++ x][y] = s[++ tot];
        while(y - 1 > 0 && !a[x][y - 1])
            a[x][-- y] = s[++ tot];
        while(x - 1 > 0 && !a[x - 1][y])
            a[-- x][y] = s[++ tot];
    }

    for(int i = 1; i <= c; i ++){
        printf("%d", a[i][1]);
        for(int j = 2; j <= r; j ++)
            printf(" %d", a[i][j]);
        printf("\n");

    }
    return 0;
}