PAT 乙级 1050（螺旋矩阵 ）

题目要求

本题要求将给定的 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>
#include<cmath>
#include<string.h>
using namespace std;
bool cmp(int a,int b)
{
    return a>b;
}
int main()
{
    int N;
    cin >> N;
    int a[N];
    for(int i=0;i<N;i++)
    {
        cin>>a[i];
    }
    int row,col;
    //获取行数，列数
    for(int i=sqrt(N);i>0;i--)
    {
        if(N%i==0)
        {
            row=N/i;
            break;
        }
    }
    col = N/row;
    sort(a,a+N);
    int result[row][col];
    memset(result,0,sizeof(result));
    int pos=0,key=N-1;
    while(!result[pos][pos])
    {
        //left to right
        for(int i=pos;i<col-pos;i++)
        {
            if(!result[pos][i])
            {
                result[pos][i]=a[key--];
            }
        }
        //top to bottom
        for(int i=pos+1;i<row-pos;i++)
        {
            if(!result[i][col-pos-1])
            {
                result[i][col-pos-1]=a[key--];
            }
        }
        //right to left
        for(int i=col-pos-1;i>=pos;i--)
        {
            if(!result[row-pos-1][i])
            {
                result[row-pos-1][i]=a[key--];
            }
        }
        //bottom to top
        for(int i=row-pos-1;i>pos;i--)
        {
            if(!result[i][pos])
            {
                result[i][pos]=a[key--];
            }
        }
        pos++;
    }
    for(int i=0;i<row;i++)
    {
        bool flag = false;
        for(int j=0;j<col;j++)
        {
            if(j==0){
                printf("%d",result[i][j]);
                flag=true;
            }
            else{
                printf(" %d",result[i][j]);
            }
        }
        if(flag){
            printf("\n");
        }
    }
    system("pause");
}

思路

本题较为复杂，主要是排列螺旋矩阵的时候比较绕，这里将矩阵划分为四个方向填充数字，分别是从左到右、从上到下、从右到左、从下到上。根据这四个方向按照一定的规律进行赋值。
值得注意的是，在对数组初始化的时候需要用到memset函数，而不能int result[row][col]={0};这样的出来的结果是有问题的，应该是在判断!result[x][x]的时候出现了某种问题，大家如果知道的话还请留言！
如此，完成。