【PAT】乙级 1050 螺旋矩阵 (25 分) c++

1050 螺旋矩阵 (25 分)

本题要求将给定的 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<iostream>
#include<vector>
#include<algorithm>
#include<cmath>
using namespace std;
bool cmp(int a,int b){return a>b;}
int main(){
	int N;
	vector<int>vnum;
	int num;
	cin>>N;
	for(int i=0;i<N;i++){
		cin>>num;
		vnum.push_back(num);
	} 
	sort(vnum.begin(),vnum.end(),cmp);
	
	int minN=sqrt(N);
	int maxN;
	for(int i=minN;i>=1;i--){
		if (N%i==0) { 
            maxN=N/i;
            minN=i;
            break;
        }
	}
	
	int arr[maxN][minN];
	fill(arr[0],arr[0]+maxN*minN,0);
	num=0;

	int layer=maxN/2+maxN%2; 
	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;

}