PAT乙级真题 1050 螺旋矩阵 C++实现

题目

本题要求将给定的 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

思路

此题分两步。

求m、n

求根号N以内最大的因子，即为n，N / n 即为m。

填充蛇形数组

和打印蛇形数组类似：Python 生成蛇形数组/对角线数组

此类问题，拐几次弯回到原来的模式，就用几层循环即可。

本文用了4层内循环，即每次填充一个圈，用round记录圈数。每次循环后判断库存是否用完了，用完后就跳出。

代码

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

int main(){
    int k;
    cin >> k;
    int n = 1;
    int m = k;
    while (n<=sqrt(k)){
        if (k%n==0){
            m = k / n;
        }
        n++;
    }
    n = k / m;

    vector<int> a(k);
    for(int i=0; i<k; i++){
        cin >> a[i];
    }
    
    sort(a.begin(), a.end());
    vector<vector<int> > b(m, vector<int>(n));

    int count = k - 1;
    int round = 0;
    int i = 0;
    int j = 0;
    while (1){
        for (j=round; j<n-round; j++){
            b[i][j] = a[count];
            count--;
        }
        if (count<0) break;

        j--;
        for (i=round+1; i<m-round; i++){
            b[i][j] = a[count];
            count--;
        }
        if (count<0) break;

        i--;
        for (j=n-round-2; j>=round; j--){
            b[i][j] = a[count];
            count--;
        }
        if (count<0) break;

        j++;
        for (i=m-round-2; i>round; i--){
            b[i][j] = a[count];
            count--;
        }
        if (count<0) break;

        i++;
        round++;
    }

    for (i=0; i<m; i++){
        cout << b[i][0];
        for (j=1; j<n; j++){
            cout << " " << b[i][j];
        }
        cout << endl;
    }
    return 0;
}