PAT 1050螺旋矩阵

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

输入格式：

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

输出格式：

输出螺旋矩阵。每行n个数字，共m行。相邻数字以1个空格分隔，行末不得有多余空格。

输入样例：

1237 76 20 98 76 42 53 95 60 81 58 93

输出样例：

98 95 9342 37 8153 20 76
58 60 76
分析：
先按照题目要求，求出m与n.将输入的数据进行排序，最后按照螺旋矩阵存放.
源代码：
#include<iostream>#include<vector>#include<algorithm>#include<cmath>using namespace std;bool cmp(int a,int b){return a>=b;}int main(){       int N;cin>>N;    vector<int> v(N);    for(int i=0;i<N;i++)    cin>>v[i];    sort(v.begin(),v.end(),cmp);    int m,n;    m=n=sqrt(N); //从平方根开始，可以自己推算    while(m*n!=N)    {    m++;    if(m*n>N)    n--;}    vector<vector<int> >matrix(m);//二维数组    for(int i=0;i<m;i++)    matrix[i]=vector<int>(n);    int num=0;    int hang1=0,hang2=m-1,lie1=0,lie2=n-1;//hang1,hang2表示行的界限，lie1,lie2表示列的界限    while(num<N)    {    for(int i=lie1;i<=lie2&&num<N;i++)    {matrix[hang1][i]=v[num];num++;    }        hang1++;       for(int i=hang1;i<=hang2&&num<N;i++)    {    matrix[i][lie2]=v[num];    num++;}lie2--;for(int i=lie2;i>=lie1&&num<N;i--){matrix[hang2][i]=v[num];num++;}    hang2--;for(int i=hang2;i>=hang1&&num<N;i--){matrix[i][lie1]=v[num];num++;}    lie1++;} for(int i=0;i<m;i++)for(int j=0;j<n;j++){cout<<matrix[i][j];if(j==n-1)cout<<endl;else cout<<" ";}return 0; }