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]
;//用二维数组表示螺旋矩阵
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;
}