PAT (Basic Level):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<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
bool cmp(int a,int b){return a>b;}
int main(){
    int N;cin>>N;int a[N],m,n;
    for(int i=0;i<N;i++) cin>>a[i];
    sort(a,a+N,cmp);
    for(int i=sqrt(N);i>0;i--){
        if(N%i==0){
            n=i;m=N/i;break;
        }
    }
    int matrix[m][n];
    int row=0,col=0,no=0,cir=0;
    while(no<N){
        matrix[row][col]=a[no++];
        if(row==cir&&col<n-1-cir) col++;
        else if(col==n-1-cir&&row<m-1-cir) row++;
        else if(row==m-1-cir&&col>cir) col--;
        else if(col==cir&&row>cir+1) row--;
        else if(col==cir&&row==1+cir){
            col++;cir++;
        }
    }
    for(int i=0;i<m;i++){
        for(int j=0;j<n-1;j++){
            cout<<matrix[i][j]<<" ";
        }
        cout<<matrix[i][n-1]<<endl;
    }
    return 0;
} 

测试结果