PAT 乙级 1050 螺旋矩阵
原创
©著作权归作者所有:来自51CTO博客作者漫浸天空的雨色的原创作品,请联系作者获取转载授权,否则将追究法律责任
1050 螺旋矩阵 (25 point(s))
本题要求将给定的 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
经验总结:
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;
}