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

1050 螺旋矩阵 (25分)

本题要求将给定的 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
作者: CHEN, Yue
单位: 浙江大学
时间限制: 200 ms
内存限制: 64 MB
代码长度限制: 16 KB

#include <iostream>
#include <vector>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
bool cmp(int a,int b){
    return a>b?true: false;
}
int main(){
    int n,col,row;
    cin>>n;
    vector<int> store(n);
    for (int i = 0; i < n; ++i) {
        cin>>store[i];
    }
    sort(store.begin(),store.end(),cmp);
    int temp = int(sqrt(n));
    for (int i = temp; i <= n; ++i) {
        int mid = n / i;
        if(mid * i == n){
            row = i > mid ? i : mid;
            col = i < mid ? i : mid;
            break;
        }
    }
    int ans[row][col];
    memset(ans,0, sizeof(ans));
    int i = 0,circle=0;
    while (i < n){
        //从左到右
        for (int j = circle; j < col - circle; ++j) {
            if(ans[circle][j] == 0){
                ans[circle][j] = store[i++];
            }
        }
        //从上到下
        for (int j = circle + 1; j < row - circle; ++j) {
            if(ans[j][col-1-circle] == 0){
                ans[j][col-1-circle] = store[i++];
            }
        }
        //从右到左
        for (int j = col - circle - 1; j > circle ; --j) {
            if(ans[row-circle-1][j] == 0)
                ans[row-circle-1][j] = store[i++];
        }
        //从下到上
        for (int j = row - circle -1; j > circle ; --j) {
            if(ans[j][circle] == 0)
                ans[j][circle] = store[i++];
        }
        circle++;
    }
    for (int i = 0; i < row; ++i) {
        for (int j = 0; j < col; ++j) {
            if(j == 0)
                cout<<ans[i][j];
            else
                cout<<" "<<ans[i][j];
        }
        if(i != row-1)
            cout<<endl;
    }
    return 0;
}

来源：CSDN

作者：亓官劼

链接：https://blog.csdn.net/qq_43422111/article/details/104107535