[PAT-乙级]1050.螺旋矩阵

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个待填充的正整数。所有数字不超过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<stdio.h>
#include<math.h>
#include<string.h>
#include<algorithm>
using namespace std;

bool cmp(int x, int y)
{
if(x != y)
return x > y;
else
return x > y;
}

int main()
{
//freopen("D://input.txt", "r", stdin);
int n;
scanf("%d", &n);

int input[n+5];
for(int i = 0; i < n; i++)
scanf("%d", &input[i]);
sort(input, input+n, cmp);

int w=0, h=0, min = n;
// 找到所要打印的高度和宽度
for(int i = 0; i <= n; i ++)
{
for(int j = 0; j <= i; j ++)
{
if(i*j == n && (i-j) < min)
{
min = i-j;
h = i;
w = j;
}
}
}

int buf[h][w];
memset(buf, 0, sizeof(buf));
int x = 0, y = 0, p = 0;
buf[x][y] = input[p ++];

while(p < n)
{
while(y+1 < w && !buf[x][y+1])//右
buf[x][++ y] = input[p ++];

while(x+1 < h && !buf[x+1][y])//下
buf[++x][y] = input[p ++];

while(x-1 >= 0 && !buf[x-1][y])//左
buf[-- x][y] = input[p ++];

while(y-1 >= 0 && !buf[x][y-1])//上
buf[x][-- y] = input[p ++];
}

for(int i = 0; i < h; i ++)
{
for(int j = 0; j < w; j ++)
{
if(j != 0)
printf(" ");
printf("%d", buf[i][j]);
}
printf("\n");
}

return 0;
}