PAT 1050. 螺旋矩阵

PAT 1050. 螺旋矩阵

本题要求将给定的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

分析

一层一层的处理

代码如下

#include<iostream>
#include<math.h>
#include<algorithm>
int getcolumn(int N){
	int t=sqrt(N);
	while(t>=1){
		if(N%t==0)
		return t;
		else t--;
	}
	return 1;
}
bool compare(const int &a,const int &b){
	return a>b;
} 
using namespace std;
int main(){
	int N; cin>>N;
	int a[N],column,row,level,t=0;
	for(int i=0;i<N;i++)
	cin>>a[i];
	sort(a,a+N,compare);
	column=getcolumn(N);
	row=N/column;
	level=row/2+row%2;
	int b[row][column]={0};
	for(int i=0;i<level;i++){
	   for(int j=i;j<=column-1-i&&t<N;j++)
	   	b[i][j]=a[t++];
	   for(int j=i+1;j<=row-2-i&&t<N;j++)
	    b[j][column-i-1]=a[t++];
	   for(int j=column-1-i;j>=i&&t<N;j--)
	    b[row-1-i][j]=a[t++];
	   for(int j=row-2-i;j>=i+1&&t<N;j--)
	    b[j][i]=a[t++];
	}
	for(int i=0;i<row;i++){
		for(int j=0;j<column-1;j++)
			cout<<b[i][j]<<" ";
			cout<<b[i][column-1]<<endl;
	}
	return 0;
}