[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个待填充的正整数。所有数字不超过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<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;
}