Pat乙级-1050. 螺旋矩阵

本题要求将给定的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<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
int cmp(int a, int b) {
  return a > b;
}
int map[10001][10001] = {0}, a[10004];
void fill(int x, int y) { // 按照螺旋顺序填充
  int i, j, k = 0;
  for (i = 1; i <= y/2; i++) {
    for (j = i; j <= y-i; j++)
      map[i][j] = a[k++];
    for (j = i; j <= x-i; j++)
      map[j][y-i+1] = a[k++];
    for (j = y-i+1; j > i; j--)
      map[x-i+1][j] = a[k++];
    for (j = x-i+1; j > i; j--)
      map[j][i] = a[k++];
  }
  if (y % 2 == 1) {
    for (i = ((y/2) + 1); i <= x-(y/2); i++)
      map[i][(y/2)+1] = a[k++];
  }
}
int main() {
  int n;
  cin >> n;
  for (int i = 0; i < n; i++)
    cin >> a[i];
  sort(a, a+n, cmp);
  int x, y;
  for(int i = 1; i <= sqrt(n); i++)
    if(n % i == 0){
      y = i;
    }
  x = n / y;
  fill(x, y);
  for (int i = 1; i <= x; i++) {
    for (int j = 1; j <= y; j++) {
      cout << map[i][j];
      if (j != y) cout << " ";
    }
    cout << endl;
  }
}