B1050 螺旋矩阵 （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

思路：

找规律，向右、向下、向左、向上输出

注意：

定义matrix[maxn][maxn]会超时，声明数组的大小和m，n不相吻合，导致数组填充的时候越界了，进而导致循环不能跳出

改进：matrix[n+1][m+1] 可以AC

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

const int maxn = 10010;

int a[maxn];

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

int main(){
    int N;
    scanf("%d", &N);
    for(int i = 0; i < N; i++){
        scanf("%d", &a[i]);
    }
    if(N == 1){
        printf("%d", a[0]);
        return 0;
    }
    sort(a, a + N, cmp);

    int m = (int)ceil(sqrt(1.0 * N));
    while(N % m != 0){
        m++;
    }
    //n为列数，m为行数
    int n = N / m, i = 1, j = 1, now = 0;
    int matrix[m + 1][n + 1];
    int U = 1, D = m, L = 1, R = n;
    while(now < N){
        while(now < N && j < R){
            matrix[i][j] = a[now++];
            j++;
        }
        while(now < N && i < D){
            matrix[i][j] = a[now++];
            i++;
        }
        while(now < N && j > L){
            matrix[i][j] = a[now++];
            j--;
        }
        while(now < N && i > U){
            matrix[i][j] = a[now++];
            i--;
        }
        U++, D--, L++, R--;
        i++, j++;
        if(now == N - 1){
            matrix[i][j] = a[now++];
        }
    }
    for(int i = 1; i <= m; i++){
        for(int j = 1; j <= n; j++){
            printf("%d", matrix[i][j]);
            if(j < n)
                printf(" ");
            else
                printf("\n");
        }
    }
    return 0;
}