1050 螺旋矩阵

1050 螺旋矩阵（25 分）

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

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

输出螺旋矩阵。每行 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<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;

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

int N ;
int main(){
    scanf("%d",&N);
    int an,am;
    int m,n;
    int min = 1000000;

    int num[N];
    for(int i=0 ;i<N ;i++){
        scanf("%d",&num[i]);
    } 

    //遍历，求出n和m 
    for(int i=1 ;i<=sqrt(N) ;i++){
        int temp = N;
        if(N%i==0){
            n = i;
            m = temp/i;
            if(m>=n&&m-n<min){
                min = m-n;
                an = n;
                am = m; 
            }else{
                continue;
            }
        }
    }
    //降序排列 
    sort(num,num+N,cmp);

    int ans[am][an];//用来存储要输出的数据 

    //赋初值-1 
    for(int i=0 ;i<am ;i++){
        for(int j=0 ;j<an ;j++){
            ans[i][j]=-1;
        }
    }


    int a1=0,a2=0;

    bool right=true,down=false,left=false,up=false;
    for(int i=0 ;i<N ;i++){
        //1 右 
        if(right){
            if(a2<an&&ans[a1][a2]==-1){
                ans[a1][a2] = num[i];
                a2++;
                continue;
            }else{
                down = true;
                right = false;
                a2--;
                a1++;
            }
        }

        //2 下 
        if(down){
            if(a1<am&&ans[a1][a2]==-1){
                ans[a1][a2] = num[i];
                a1++;
                continue;
            }else{
                down = false;
                left = true;
                a1--;
                a2--;   
            }   
        }

        //3 左 
        if(left){
            if(a2>=0&&ans[a1][a2]==-1){
                ans[a1][a2] = num[i];
                a2--;
                continue;
            }else{
                left = false;
                up = true;
                a2++;
                a1--;
            }
        }

        //4 上 
        if(up){
            if(a1>=0&&ans[a1][a2]==-1){
                ans[a1][a2] = num[i];
                a1--;
                continue;
            }else{
                up = false;
                right = true;
                a1++;
                a2++;       
            }   
        }
        //再跟一个右，如果第一次up==false right==true更新后
        //因为写的是顺序语句，所以不会返回去进行判断，直接进行下一次循环，导致结果出现错误。 
        //1 右 
        if(right){
            if(a2<an&&ans[a1][a2]==-1){
                ans[a1][a2] = num[i];
                a2++;
                continue;
            }else{
                down = true;
                right = false;
                a2--;
                a1++;
            }
        }

    }


    for(int i=0 ;i<am ;i++){
        for(int j=0 ;j<an ;j++){
            if(j==0){
                printf("%d",ans[i][j]);
            }else{
                printf(" %d",ans[i][j]);
            }
        } 
        printf("\n");
    }

    return 0;}