【螺旋矩阵】-PAT

螺旋矩阵

本题要求将给定的 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 <iostream>
#include <algorithm>
using namespace std;
bool cmp(int a,int b)
{
    return a>b;
}
void maxmn(int N,int &m,int &n)//确定题目规定的行数和列数，和判断素数相似
{
    int min=100001;
    for(int i=1;i<=N;i++)
    {
        for(int j=1;j<=N;j++)
        {
            if(i*j==N)
            {
                while(i>=j&&i-j<min)
                {
                    min=i-j;
                    m=i;
                    n=j;
                }
            }
        }
    }
}
int main()
{
    int N,m,n,i,j,m1,n1,k=0;
    cin>>N;
    int *a=(int*)malloc(sizeof(int)*N);//创建一维数组
    for(i=0;i<N;i++)
    {
        int p;
        cin>>p;
        a[i]=p;
    }
    sort(a,a+N,cmp);//按非递增顺序排列数组
    maxmn(N,m,n);
    m1=m;//变量改变，后期输出需要
    n1=n;//同上
    int **mat=new int*[m];//创建二维数组
    for(i=0;i<m;i++)
    {
        mat[i]=new int[n];
    }
    bool right=1,down=1;//right控制向右，down控制向下
    i=0;j=-1;//i为行变量，j为列变量
    while(m&&n)
    {
        for(int cnt=1;cnt<=n;cnt++,k++)
        {
            right?j++:j--;//向右为真，则列变量++
            mat[i][j]=a[k];
        }
        right?right=0:right=1;//旋转填充先右再下再左再上
        m--;
        for(int cnt=1;cnt<=m;cnt++,k++)
        {
            down?i++:i--;//向下为真，则行变量++
            mat[i][j]=a[k];
        }
        down?down=0:down=1;//先向下再向上
        n--;
    }
    for(i=0;i<m1;i++)
    {
        for(j=0;j<n1;j++)
        {
            printf("%d%c",mat[i][j],j==n1-1?'\n':' ');
        }
    }
    return 0;
}