PAT乙级-1050 螺旋矩阵(25 分)

题目链接

#include<iostream>
#include<cmath>
#include<vector>
#include<algorithm>
using namespace std;

bool cmp(int a,int b) {
    return a>b;
}

int main(){
    int N,m,n;
    cin>>N;
    for(n=(int)sqrt(N*1.0);n>=1;n--)
        if(N%n==0) break;
    m=N/n;
    vector<int> v(N);
    for(int i=0;i<N;i++) cin>>v[i];
    sort(v.begin(),v.end(),cmp);
    vector<vector<int>> ans(m,vector<int>(n));
    int level=m/2+m%2,cnt=0; //注意level
    for(int i=0;i<level;i++){ //注意加上cnt<N
        for(int j=i;j<n-i&&cnt<N;j++) ans[i][j]=v[cnt++]; //行固定，列变化
        for(int j=i+1;j<m-i&&cnt<N;j++) ans[j][n-i-1]=v[cnt++]; //列固定，行变化
        for(int j=n-i-2;j>=i&&cnt<N;j--) ans[m-i-1][j]=v[cnt++]; //行固定，列变化
        for(int j=m-i-2;j>=i+1&&cnt<N;j--) ans[j][i]=v[cnt++]; //列固定，行变化
    }
    for(int i=0;i<m;i++){
        for(int j=0;j<n;j++){
            if(j!=0) cout<<" ";
            cout<<ans[i][j];
        }
        cout<<endl;
    }
    return 0;
}