1050 螺旋矩阵 (25分)C++示例代码

1050 螺旋矩阵 (25分)

本题要求将给定的 N 个正整数按非递增的顺序，填入“螺旋矩阵”。所谓“螺旋矩阵”，是指从左上角第 1 个格子开始，按顺时针螺旋方向填充。要求矩阵的规模为 $m$ 行 $n$列，满足条件：$m\times n$ 等于 $N$；$m\ge 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

C++示例代码：

#include
#include
#include
#include
using namespace std;
bool cmp(int a,int b){return a>b;}
int main(){
	int N;
	vector<int>vnum;
	int num;
	scanf("%d",&N);
	for(int i=0;i<N;i++){
		scanf("%d",&num);
		vnum.push_back(num);
	} 
	sort(vnum.begin(),vnum.end(),cmp);//降序排序 
	
	int minN=sqrt(N);//n 
	int maxN;//m
	for(int i=minN;i>=1;i--){//计算出maxN和minN 
		if (N%i==0) { 
            maxN=N/i;
            minN=i;
            break;
        }
	}
	
	int arr[maxN][minN];
	fill(arr[0],arr[0]+maxN*minN,0);//填充0 
	num=0;

	int layer=maxN/2+maxN%2;//需要绕layer圈 //解决测试点超时问题 
	for(int k=0;k<layer;k++){
		int min=minN-1-k;//定位该圈右下角的列坐标 
		int max=maxN-1-k;//定位该圈右下角的行坐标 
		for(int i=k;i<=min&&num<N;i++)//遍历一整行 
			arr[k][i]=vnum[num++];
		
		for(int i=k+1;i<=max-1&&num<N;i++)//该列遍历少两个 
			arr[i][min]=vnum[num++];
		
		for(int i=min;i>=k&&num<N;i--)//遍历一整行 
			arr[max][i]=vnum[num++];
		
		for(int i=max-1;i>=k+1&&num<N;i--)//该列遍历少两个 
			arr[i][k]=vnum[num++];
	}		
	
	for(int j=0;j<maxN;j++){
		for(int i=0;i<minN;i++){
			if(i!=0)printf(" ");
			printf("%d",arr[j][i]);
		}
		printf("\n");
	}	
	return 0;

}