PAT乙级 1050. 螺旋矩阵(25)

1050. 螺旋矩阵(25)

本题要求将给定的N个正整数按非递增的顺序，填入“螺旋矩阵”。所谓“螺旋矩阵”，是指从左上角第1个格子开始，按顺时针螺旋方向填充。要求矩阵的规模为m行n列，满足条件：m*n等于N；m>=n；且m-n取所有可能值中的最小值。

输入格式：

输入在第1行中给出一个正整数N，第2行给出N个待填充的正整数。所有数字不超过10⁴，相邻数字以空格分隔。

输出格式：

输出螺旋矩阵。每行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

时间限制

150 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

CHEN, Yue

分析：

1. 确定m和n的值。题目规定m*n=N， m >= n， m-n尽可能小，所以可以从大于N的平方根开始寻找m的值，只需要令N % m == 0即可，N / m即为n的值。

2. 生成螺旋矩阵，用二维数组储存比较直观。可以用left，right，top， bottom来标记每次的边界。由于数组的大小N已知，故可以此作为停止条件。

#include <iostream>
#include <algorithm>
#include <vector>
#include <math.h>
using namespace std;
bool cmp(int a, int b) {
    return a > b;
}

int main() {
    int N;
    cin >> N;
    vector<int> a(N);
    for (int i = 0; i < N; i++) {
        cin >> a[i];
    }
    sort(a.begin(), a.end(), cmp);
    double sqr = sqrt(N);
    int sqrN = ceil(sqr); // 向上取整 

    int m, n, k = 0;
    for (int i = sqrN; i <= N; i++) {
        if (N % i == 0) {
            m = i;
            n = N / m;
            break;
        }
    }

    vector<vector<int> > b(m, vector<int>(n));
    int left = 0, right = n - 1;
    int top = 0, bottom = m - 1;
    while (k < N) {
        for (int i = left; i <= right; i++) b[top][i] = a[k++];
        for (int j = top + 1; j <= bottom; j++) b[j][right] = a[k++];
        for (int i = right - 1; i >= left; i--) b[bottom][i] = a[k++];
        if (right > left) {
            for (int j = bottom - 1; j > top; j--) b[j][left] = a[k++];
        }
        left++;
        top++;
        right--;
        bottom--;
    }
    for (int i = 0; i < m; i++) {
        for (int j = 0; j < n; j++) {
            cout << b[i][j];
            if (j != n - 1) cout << " ";
        }
        cout << endl;
    }
    return 0;
}