PAT乙级1050. 螺旋矩阵(25)

题目：

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

输入格式：

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

输出格式：

输出螺旋矩阵。每行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<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
int main() {
    int N, m, n, i, j, t=0;
    cin>>N;
    int a[N];
    for (i=0; i<N; i++)
        cin>>a[i];
    sort (a, a+N, greater_equal<int>());
    for (i=sqrt(N); i>0; i--) {
        if (N%i == 0) {
            n = i;
            m = N/n;
            break;
        }
    }
    int b[m][n]={0};
    int level = m/2 + m%2;
    for (i=0; i<level; i++) {
        for (j=i; j<=n-1-i && t<=N-1; j++) {
            b[i][j] = a[t++];
        }
        for (j=i+1; j<=m-2-i && t<=N-1; j++) {
            b[j][n-1-i] = a[t++];
        }
        for (j=n-i-1; j>=i && t<=N-1; j--) {
            b[m-1-i][j] = a[t++];
        }
        for (j=m-2-i; j>=i+1 && t<=N-1; j--) {
            b[j][i] = a[t++];
        }
    }
    for (i=0; i<m; i++) {
        for (j=0; j<n; j++) {
            cout<<b[i][j];
            if (j != n-1)
                cout<<" ";
        }
        cout<<endl;
    }
    return 0;
}