PAT 乙级 1050 螺旋矩阵

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/a845717607/article/details/86168578

1050 螺旋矩阵 （25 point(s)）

本题要求将给定的 N 个正整数按非递增的顺序，填入“螺旋矩阵”。所谓“螺旋矩阵”，是指从左上角第 1 个格子开始，按顺时针螺旋方向填充。要求矩阵的规模为 m 行 n 列，满足条件：m×n 等于 N；m≥n；且 m−n 取所有可能值中的最小值。

输入格式：

输入在第 1 行中给出一个正整数 N，第 2 行给出 N 个待填充的正整数。所有数字不超过 10​4​​，相邻数字以空格分隔。

输出格式：

输出螺旋矩阵。每行 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

经验总结：

emmmmmm 怎么说呢，这一题其实思路上还是有点难度的，不知道怎么控制输出，实际上解决方法是按照顺序控制下标一点点的填满二维数组，最后再按照二维数组的规律输出，详细代码就在下面，这里说一点，书上的代码提交后，最后一个测试点会超时（或者出错？），我的解决方法是，对于所有的10^4以下的质数进行特判，因为质数N肯定是N行1列，如果用传统的思路可能就会超时，特判的方式当然就是利用质数表啦~(๑•̀ㅂ•́)و✧

AC代码 

#include <cstdio>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn=10010;
int a[maxn],b[maxn][maxn];
bool flag[maxn]={false};
bool cmp(int a,int b)
{
	return a>b;
}
void find_prime()
{
	for(int i=2;i<maxn;++i)
	{
		if(flag[i]==false)
		{
			for(int j=i+i;j<maxn;j+=i)
			{
				flag[j]=true;
			}
		}
	}
}
void print(int row,int col)
{
	for(int i=1;i<=row;++i)
	{
		for(int j=1;j<=col;++j)
		{				
			printf("%d",b[i][j]);
			if(j!=col)
				printf(" ");
			else
				printf("\n");
		}
	}
}
int main()
{
	int n;
	find_prime();
	while(~scanf("%d",&n))
	{
		for(int i=0;i<n;++i)
			scanf("%d",&a[i]);
		if(n==1)
		{
			printf("%d",a[0]);
		}
		else if(flag[n]==false)
		{
			sort(a,a+n,cmp);
			for(int i=0;i<n;++i)
				printf("%d\n",a[i]);
		}
		else
		{

			sort(a,a+n,cmp);
			int row=(int)ceil(sqrt(1.0*n));
			while(n%row!=0)
				++row;
			int col=n/row;
			int U=1,D=row,L=1,R=col;
			int i=1,j=1,num=0;
			while(num<n)
			{
				while(j<R&&num<n)
				{
					b[i][j]=a[num++];
					++j;
				}
				while(i<D&&num<n)
				{
					b[i][j]=a[num++];
					++i;
				}
				while(j>L&&num<n)
				{
					b[i][j]=a[num++];
					--j;
				}
				while(i>U&&num<n)
				{
					b[i][j]=a[num++];
					--i;
				}
				++U,--R,--D,++L;
				++j;++i;
				if(num==n-1)
				{
					b[i][j]=a[num++];
				}
			}
			print(row,col);
		}
	}
	return 0;
}