1050. 螺旋矩阵(25)

1050. 螺旋矩阵(25)

时间限制

150 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

CHEN, Yue

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

#include <stdio.h>
#include <iostream>
#include <stack>
#include <string.h>
#include <queue>
#include <cmath>
#include <vector>
#include <algorithm>
#include <map>
#include <set>
#include <string>
using namespace std;
typedef long long LL;
bool cmp(int a, int b)
{
    return a > b;
}
int arr[10000][10000];
//int a[100000];
int main() {
    //freopen("in.txt", "r", stdin);
    //freopen("out.txt","w",stdout);
    int num;
    cin >> num;
    vector<int> a(num);
    for(int i = 0; i < num; i++){
        cin >> a[i];
    }
    sort(a.begin(), a.end(), cmp);
    int last_n = 0, last_m = 0;
    int min = 65535;
    for(int i = 0; i <= num; i++){
        for(int j = 0; j <= num; j++){
            if(i * j == num && i >= j && i - j < min){
                last_m = i;
                last_n = j;
                min = i - j;
            }

        }
    }
    int index = 0;
    int level = last_m  / 2; //m行n列 此时m = 4, n = 3
    if(last_m % 2 != 0){
        level++; //此时level = 2
    }
    for(int i = 0; i < level; i++){
        for(int j = i; j <= last_n - 1 - i && index <= num - 1; j++){
            arr[i][j] = a[index++];
        }
        for(int j = i + 1; j <= last_m - 2 - i && index <= num - 1; j++){
            arr[j][last_n - 1 - i] = a[index++];
        }
        for(int j = last_n - i - 1; j >= i && index <= num - 1; j--){
            arr[last_m - 1 - i][j] = a[index++];
        }
        for(int j = last_m - 2 - i; j >= i + 1 && index <= num - 1; j--){
            arr[j][i] = a[index++];
        }
    }
    for(int i = 0; i < last_m; i++){
        for(int j = 0; j < last_n; j++){
            printf("%d", arr[i][j]);
            if(j != last_n - 1){
                printf(" ");
            }
        }
        printf("\n");
    }

    //cout << last_m << " " << last_n << endl;

    return 0;
}