PAT甲级真题1105 螺旋矩阵（两种方法）

给定一个包含 NN 个正整数的序列，请你将序列中的元素以非递增顺序填充到螺旋矩阵中。

从左上角的第一个元素开始填充，并按照顺时针方向旋转。

要求矩阵有 mm 行 nn 列，并且 m,nm,n 满足：

m×n=Nm×n=N
m≥nm≥n
m−nm−n 尽可能小
输入格式
第一行包含整数 NN。

第二行包含 NN 个整数。

输出格式
输出填充好的 m×nm×n 的矩阵。

数字之间用一个空格隔开，结尾不得有多余空格。

数据范围
1≤N≤100001≤N≤10000,
1≤1≤ 序列中元素 ≤10000≤10000
输入样例：
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 <cstring>
#include <algorithm>
#include <vector>

using namespace std;

const int N = 10010;

int n;
int w[N];

int main()
{
    cin >> n;
    for (int i = 0; i < n; i ++ ) cin >> w[i];

    sort(w, w + n, greater<int>());//从大到小排序

    int r, c;
    for (int i = 1; i <= n / i; i ++ )
        if (n % i == 0)
        {
            r = n / i;
            c = i;
        }

    vector<vector<int>> res(r, vector<int>(c));

    int dx[] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1};
    for (int i = 0, x = 0, y = 0, d = 1; i < n; i ++ )
    {
        res[x][y] = w[i];
        int a = x + dx[d], b = y + dy[d];
        if (a < 0 || a >= r || b < 0 || b >= c || res[a][b])
        {
            d = (d + 1) % 4;
            a = x + dx[d], b = y + dy[d];
        }
        x = a, y = b;
    }

    for (int i = 0; i < r; i ++ )
    {
        cout << res[i][0];
        for (int j = 1; j < c; j ++ )
            cout << ' ' << res[i][j];
        cout << endl;
    }

    return 0;
}

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>

using namespace std;

const int N = 10010;

int n;
int w[N];

int main()
{
    cin >> n;
    for (int i = 0; i < n; i ++ ) cin >> w[i];

    sort(w, w + n, greater<int>());

    int r, c;
    for (int i = 1; i <= n / i; i ++ )
        if (n % i == 0)
        {
            r = n / i;//行
            c = i;
        }

    vector<vector<int>> res(r, vector<int>(c));

    int cnt=1,x=0,y=0;
    res[0][0]=w[0];//必须先填
    while(cnt<n){
        while(y+1<c&&!res[x][y+1]&&cnt<n)res[x][++y]=w[cnt++];
        while(x+1<r&&!res[x+1][y]&&cnt<n)res[++x][y]=w[cnt++];
        while(y-1>=0&&!res[x][y-1]&&cnt<n)res[x][--y]=w[cnt++];
        while(x-1>=0&&!res[x-1][y]&&cnt<n)res[--x][y]=w[cnt++];
    }

    for (int i = 0; i < r; i ++ )
    {
        cout << res[i][0];
        for (int j = 1; j < c; j ++ )
            cout << ' ' << res[i][j];
        cout << endl;
    }

    return 0;
}