PAT-A-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<iostream>
#include<cstdio>
#include<math.h>
#include<algorithm>

using namespace std;
const int maxn = 10010;

int a[maxn];
int b[maxn][maxn];

int cmp(int a, int b)
{
  return a > b;
}
int main()
{
  int num;
  cin >> num;
  for (int i = 0; i < num; i++)
    cin >> a[i];

  if (num == 1)
  {
    cout << a[0];
    return 0;
  }

  int sqr = (int)ceil(sqrt(1.0*num));

  int m = sqr;
  while (num % m != 0)
  {
    m++;
  }
  int n = num / m;
  sort(a, a + num, cmp);

  int now = 0;

  int u = 1, d = m, l = 1, r = n;
  int i = 1, j = 1;
  while (now < num)
  {
    while (j < r&&now < num)
    {
      b[i][j] = a[now];
      now++;
      j++;
    }

    while (i < d&&now < num)
    {
      b[i][j] = a[now];
      now++;
      i++;
    }

    while (j>l&&now < num)
    {
      b[i][j] = a[now];
      now++;
      j--;
    }

    while (i>u&&now < num)
    {
      b[i][j] = a[now];
      now++;
      i--;
    }

    r--;
    l++;
    u++;
    d--;
    i++;
    j++;
    
    if (now == num - 1)
    {
      b[i][j] = a[now];
      now++;
    }

  }

  for (int i = 1; i <= m; i++)
  {
    for (int j = 1; j <= n; j++)
    {
      cout << b[i][j];

      if (j != n)
        cout << " ";

    }
    cout << endl;
  }

  system("pause");
  return 0;
}