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

本题要求将给定的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

思路：首先根据题意用双层for循环遍历出符合要求的m和n值，然后对输的N个数据排序，一会螺旋放数的时候方便。放数规则请看注释。

#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;}

题目链接：

https://www.patest.cn/contests/pat-b-practise/1050