1050 螺旋矩阵

1050 螺旋矩阵

本题要求将给定的 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<cmath>
#include<vector>
#include<algorithm>
using namespace std;

int main(){
    int n;
    cin>>n;
    int listNums[n];
    for(int i=0;i<n;i++){
        cin>>listNums[i];
    }
    
    sort(listNums,listNums+n,[](auto a,auto b){
        return a>b;
    });
    
    int j,k;
    k=int(sqrt(n*1.0));
    while(n%k!=0){
        k--;
    }
    j=n/k;
//     int juzhen[j][k];
    vector<vector<int>> juzhen(j,vector<int>(k));//定义螺旋矩阵
    int shang=0,zuo=0,you=k-1,xia=j-1;//定义上下左右的边界
    int x=0,y=-1,direction=0;//初始化坐标与方向
    int d[4][2]={//定义方向
        0,1,
        1,0,
        0,-1,
        -1,0
    };
    for(int i=0;i<n;i++){
        x+=d[direction][0];
        y+=d[direction][1];
        juzhen[x][y]=listNums[i];
        if(direction==0&&y==you){//向右到达右边界，转向向下，更新上边界
            direction=(direction+1)%4;
            shang++;
        }else if(direction==1&&x==xia){//向下到达下边界，转向向左，更新右边界
            direction=(direction+1)%4;
            you--;
        }else if(direction==2&&y==zuo){//向左到达左边界，转向向上，更新下边界
            direction=(direction+1)%4;
            xia--;
        }else if(direction==3&&x==shang){//向上到达上边界，转向向右，更新左边界
            direction=(direction+1)%4;
            zuo++;
        }
    }
    for(int i=0;i<j;i++){//打印结果
        for(int u=0;u<k;u++){
            if(u!=0)cout<<" ";
            cout<<juzhen[i][u];
        }
        cout<<endl;
    }
}