pat-basic-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<cstdio>
#include<algorithm>
#include<math.h>
bool cmp(int a,int b)
{
	return a>b;
}
using namespace std;
int bn[10001][10001];
int an[10010];
int main()
{
	int N,i,n,m,j,k,cnt;
	k=0,cnt=0,i=0,j=0;
	scanf("%d",&N);
	for(i=0;i<N;i++)
	{
		scanf("%d",&an[i]);
	}
	sort(an,an+N,cmp);
	n=sqrt(N);
	while(N%n!=0)
	{
		n--;
	}
	m=N/n;
	while(1)
	{
		for(i=k,j=k;i<n-k;i++)
		{
			bn[j][i]=an[cnt];
			cnt++;
			if(cnt==N)
				goto out;
		}
		for(j=k+1,i=n-k-1;j<m-k;j++)
		{
			bn[j][i]=an[cnt++];
			if(cnt==N)
				goto out;
		}
		for(i=n-k-2,j=m-k-1;i>k;i--)
		{
			bn[j][i]=an[cnt++];
			if(cnt==N)
				goto out;
		}	
		for(j=m-k-1,i=k;j>k;j--)
		{
			bn[j][i]=an[cnt++];
			if(cnt==N)
				goto out;
		}
		k++;
	}
out:
	for(j=0;j<m;j++)
	{
		for(i=0;i<n;i++)
		{
			printf("%d",bn[j][i]);
			if(i!=n-1)
				putchar(' ');
			if(i==n-1&&j!=m-1)
				putchar('\n');
		}
	}
	return 0;
}