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

题目：

本题要求将给定的N个正整数按非递增的顺序，填入“螺旋矩阵”。所谓“螺旋矩阵”，是指从左上角第1个格子开始，按顺时针螺旋方向填充。要求矩阵的规模为m行n列，满足条件：m*n等于N；m>=n；且m-n取所有可能值中的最小值。
输入格式：
输入在第1行中给出一个正整数N，第2行给出N个待填充的正整数。所有数字不超过104，相邻数字以空格分隔。
输出格式：
输出螺旋矩阵。每行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<algorithm>
#include<cmath>
#include<vector>
#include<iostream>
using namespace std;
bool cmp(int a, int b)
{
    return a > b;//将数组按降序排列
}
int find(int x)//找矩阵长度
{
    int i = 0;
    int b = sqrt((double)x);
    while (b >= 1)
    {
        if (x %b == 0)
            return b;
        b--;
    }
    return 1;
}
int main()
{
    int N;
    cin >> N;
    int len = find(N);
    int high = N / len;
    vector<int> n(N);
    for (int i = 0; i < N; i++)
    {
        cin >> n[i];
    }
    sort(n.begin(), n.end(), cmp);
    vector<vector<int> > b(high, vector<int>(len));
    int level = high / 2 + high % 2;
    int t = 0;
    for (int i = 0; i < level; i++)
    {
        for (int j = i; j <= len - 1 - i && t <= N - 1; j++) {
            b[i][j] = n[t++];
        }
        for (int j = i + 1; j <= high - 2 - i && t <= N - 1; j++) {
            b[j][len - 1 - i] = n[t++];
        }
        for (int j = len - i - 1; j >= i && t <= N - 1; j--) {
            b[high - 1 - i][j] = n[t++];
        }
        for (int j = high - 2 - i; j >= i + 1 && t <= N - 1; j--) {
            b[j][i] = n[t++];
        }
    }
    for (int i = 0; i < high; i++) {
        for (int j = 0; j < len; j++) {
            printf("%d", b[i][j]);
            if (j != len - 1) printf(" ");
        }
        printf("\n");
    }
    return 0;
}