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<cstdio>
#include<algorithm>
#include<iostream>
#include<cmath>
using namespace std;
bool cmp(int a,int b)
{
    return a>b;
}
int main()
{
    int n,i,m,k;
    cin>>k;
    int a[k];
    for(i=0;i<k;i++)
        scanf("%d",&a[i]);
    sort(a,a+k,cmp);
    m=ceil(sqrt(k));//上取整
    n=k/m;
    while(1)//确定矩阵的规模
    {
        if(m*n!=k)m++,n=k/m;
        else
            break;
    }
    int s[m][n],x=0,j=0,c=0;//x表示排了几层，c表示数组a的下标
    i=0;
    while(c<k)
    {
        for(j=x;j<n-x;j++)//从左到右
        {
            s[i][j]=a[c++];
            if(j==n-x-1){i++;break;}
        }
        if(c==k)break;
        for(;i<m-x;i++)//从上到下
        {
            s[i][j]=a[c++];
            if(i==m-x-1){j--;break;}

        }
        if(c==k)break;
        for(;j>=x;j--)//从右到左
        {
            s[i][j]=a[c++];
            if(j==x){i--;break;}
        }
        if(c==k)break;
        for(;i>x;i--)//从下到上
        {
            s[i][j]=a[c++];
            if(i==x+1){j++;break;}
        }
        x++;
    }
    for(i=0;i<m;i++)
    {
        cout<<s[i][0];
        for(j=1;j<n;j++)
        {
            cout<<' '<<s[i][j];
        }
        cout<<endl;
    }
    return 0;
}