1050. 螺旋矩阵(25) C++

本题要求将给定的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<bits/stdc++.h>
using namespace std;
bool cmp(int a,int b)
{
	return a>b;
}
int res[10010][10010];
int main(){
	int n,i; cin>>n;
	int col = sqrt(n), row;
    while(n % col){
         col--;
     }
     row = n / col;
	int a[n];
	for(i=0;i<n;i++) cin>>a[i];
	sort(a,a+n,cmp);
 	int f=0,k=0,p=0,q=0;
			while(k<n)
			{	
				if(f%4==0)
				{
					for(i=p;i<col-p;i++)
					{
						if(a[k]>res[p][i])
							res[p][i]=a[k++];
					}
					f++;
				}
				if(k>=n) break;
				if(f%4==1)
				{
					for(i=p;i<row-p;i++)
					{
						if(a[k]>res[i][col-p-1])
							res[i][col-p-1]=a[k++];
					}
					f++;
				}
				if(k>=n) break;
				if(f%4==2)
				{
					for(i=col-p-1;i>=0;i--)
					{	
						if(a[k]>res[row-p-1][i])
							res[row-p-1][i]=a[k++];
					}
					f++;
				}
				if(k>=n) break;
				if(f%4==3)
				{
					for(i=row-p-1;i>=0;i--)
					{	
						if(a[k]>res[i][p])
							res[i][p]=a[k++];
					}
					f++;
				}
				p++;
				if(k>=n) break;			
			}	
			for(i=0;i<row;i++)
			{
				cout<<res[i][0];
				for(int j=1;j<col;j++)
					cout<<" "<<res[i][j];
				cout<<endl;		
			}    	
	 
    return 0;
}