本题要求将给定的 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
思路:首先计算矩阵的行数m和列数n。由于m>=n,且m*n=N。所以可以让行数m从sqrt(N)(向上取整)开始寻找直到N可以被m整除为止。 接下来对数据降序排序,然后设置四个边界,一圈一圈把数据保存在二维数组中,在按要求输出。注意如果N==1需要特判,另外,当N为完全平方数时,方阵最中间只有一个数,需要单独判断,赋值。 新版测试最后一组总是超时!,二维数组改用int ans[m][n],这种方式定义可以AC,但vs2010不支持这种定义。
参考代码:
#include<cstdio>
#include<cmath>
#include<vector>
#include<cstring>
#include<algorithm>
using namespace std;
int N,m,n=0,a[10010];
bool cmp(const int& a,const int& b){
return a>b;
}
int main()
{
scanf("%d",&N);
for(int i=0;i<N;i++)
scanf("%d",&a[i]);
if(N==1) {
printf("%d\n",a[0]);
return 0;
}
sort(a,a+N,cmp);
m=ceil(sqrt(1.0*N));
while(m<N&&N%m!=0) m++;
n=N/m;
int ans[m][n];
memset(ans,0,sizeof(ans));
int up=0,down=m-1,left=0,right=n-1,k=0,i=0,j=0;
while(k<N){
while(k<N&&j<right) ans[up][j++]=a[k++];
while(k<N&&i<down) ans[i++][right]=a[k++];
while(k<N&&j>left) ans[down][j--]=a[k++];
while(k<N&&i>up) ans[i--][left]=a[k++];
up++;down--;left++;right--;i=up;j=left;
if(k==N-1) ans[i][j]=a[k++];
}
for(int i=0;i<m;i++){
for(int j=0;j<n;j++){
if(j==0)
printf("%d",ans[i][j]);
else
printf(" %d",ans[i][j]);
}
printf("\n");
}
return 0;
}