B1050 螺旋矩阵 (25分)

/*
题意：
1、给出N个数，按非递增序列组成螺旋矩阵，左上角第一个格子，顺时针，
m*n = N，m>=n，而且m-n尽可能大，也就是要求金坑你方阵

解题：
1、输入N
2、求得n和m（开根号）
3、然后读入所有的元素
4、排序
5、创建二维数组
6、遍历 排序的终止条件都跟i有关

learn && wrong:
1、遍历
2、level，也就是中间值的学习
3、vector指定大小，括号（）里面加数字

*/

#include <iostream>
#include <algorithm>
#include <cmath>
#include <vector>
using namespace std;
int cmp(int a, int b) { return a > b; } //排序
int main() {
    int N, m, n, t = 0;
    scanf("%d", &N);
    for (n = sqrt((double)N); n >= 1; n--) {      //这个不错，从根号开始找
        if (N % n == 0) {   //m行n列
            m = N / n;
            break;
        }
    }
    vector<int> a(N);                             //初始化向量长度
    for (int i = 0; i < N; i++)                   //读入长度
        scanf("%d", &a[i]);
    sort(a.begin(), a.end(), cmp);
    vector<vector<int> > b(m, vector<int>(n));
    int level = m / 2 + m % 2;                    //这个特别重要，对于图形来说，层数
    for (int i = 0; i < level; i++) {
        for (int j = i; j <= n - 1 - i && t <= N - 1; j++)      //n
            b[i][j] = a[t++];
        for (int j = i + 1; j <= m - 2 - i && t <= N - 1; j++)
            b[j][n - 1 - i] = a[t++];
        for (int j = n - i - 1; j >= i && t <= N - 1; j--)
            b[m - 1 - i][j] = a[t++];
        for (int j = m - 2 - i; j >= i + 1 && t <= N - 1; j--)
            b[j][i] = a[t++];
    }
    for (int i = 0; i < m; i++) {
        for (int j = 0; j < n; j++) {
            printf("%d", b[i][j]);
            if (j != n - 1) printf(" ");
        }
        printf("\n");
    }
    return 0;
}