PAT乙级（Basic Level）1050（C++）

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<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
int main(){
    int N;
    cin>>N;
    int num[N+5];
    for(int i=0;i<N;i++)
        cin>>num[i];
    sort(num,num+N,greater<int>());
    int m=0,n=0,min=N;
    for(int i=0;i<=N;i++){
        for(int j=0;j<=i;j++){
            if(i*j==N&&(i-j)<min){//在符合条件的数中寻找差最小的数，并记录下来
                min=i-j;
                m=i;
                n=j;
            }
        }   
    }
    int screw[m][n];//用二维数组表示螺旋矩阵
    memset(screw,0,sizeof(screw));
    int i=0,j=0,k=0;
    screw[i][j]=num[k++];//首先为了以后赋值方便先将第一个元素赋值
    while(k<N){     //需要赋值N次
        while(j+1<n&&!screw[i][j+1])    screw[i][++j]=num[k++];//先往右，如果小于边界值并且没有赋值过，则赋值
        while(i+1<m&&!screw[i+1][j])    screw[++i][j]=num[k++];//再往下，如果小于边界值并且没有赋值过，则赋值
        while(i-1>=0&&!screw[i-1][j])   screw[--i][j]=num[k++];//再往左，如果小于边界值并且没有赋值过，则赋值
        while(j-1>=0&&!screw[i][j-1])   screw[i][--j]=num[k++];//再往上，如果小于边界值并且没有赋值过，则赋值
    }//至于为什么要用++j，因为如果是j++，往下的时候，j需要减一，造成了后续的繁琐
    for(int i=0;i<m;i++){
        for(int j=0;j<n;j++){
            if(j!=0)
                cout<<" ";  //如果不是第一个元素，则元素前面应有空格
            cout<<screw[i][j];
        }       
            cout<<endl;
    }
    return 0;
}