1050.螺旋矩阵

1. 螺旋矩阵(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个待填充的正整数。所有数字不超过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 <iostream>
#include <vector>
#include <algorithm> 
#include <memory.h>
using namespace std;

bool cmp(int a, int b) { return a > b; }

int func(int N) 
{
    int i = sqrt((double)N);
    while(i >= 1) {
        if (N % i == 0) 
            return i;
        i--;
    }
    return 1;
}

int main()
{
    int N, m, n;
    cin >> N;
    n = func(N);
    m = N / n;
    vector <int> a(N);
    int b[m][n];
    memset(b, -1, sizeof(b));
    for (int k = 0; k < N; k++) {
        cin >> a[k];
    }
    sort(a.begin(), a.begin() + N, cmp); 
    int total = 0;
    int i = 0, j = 0;
    while (total < N) {
            while (j <= n-1 && b[i][j] == -1) { //right
                b[i][j] = a[total++];
                j++;
            }
            i++;
            j--;
            while (i <= m-1 && b[i][j] == -1) { //down
                b[i][j] = a[total++];
                i++;
            }
            i--;
            j--;
            while (j >= 0 && b[i][j] == -1) { //left
                b[i][j] = a[total++];
                j--;
            }
            j++;
            i--;
            while (i >= 0 && b[i][j] == -1) { //up
                b[i][j] = a[total++];
                i--;
            }
            i++;
            j++;        
    }
    for (int k = 0; k < m; k++) {
        for (int l = 0; l < n; l++) {
            cout << b[k][l];
            if (l != n - 1) cout << " ";
        }
        cout << endl;
    }
    return 0;
}