PAT 1050螺旋矩阵

本题要求将给定的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
分析：
先按照题目要求，求出m与n.将输入的数据进行排序，最后按照螺旋矩阵存放.
源代码：
#include<iostream>
#include<vector>
#include<algorithm>
#include<cmath>
using namespace std;
bool cmp(int a,int b)
{
	return a>=b;
}
int main()
{   
    int N;cin>>N;
    vector<int> v(N);
    for(int i=0;i<N;i++)
    cin>>v[i];
    sort(v.begin(),v.end(),cmp);
    int m,n;
    m=n=sqrt(N); //从平方根开始，可以自己推算
    while(m*n!=N)
    {
    	m++;
    	if(m*n>N)
    	n--;
	}
    vector<vector<int> >matrix(m);//二维数组
    for(int i=0;i<m;i++)
    matrix[i]=vector<int>(n);
    int num=0;
    int hang1=0,hang2=m-1,lie1=0,lie2=n-1;//hang1,hang2表示行的界限，lie1,lie2表示列的界限
    while(num<N)
    {    
		for(int i=lie1;i<=lie2&&num<N;i++)
    	{
		matrix[hang1][i]=v[num];
			num++;
	    }
	    
	    hang1++;
	   
	   
		 for(int i=hang1;i<=hang2&&num<N;i++)
	    {
	    	matrix[i][lie2]=v[num];
	    	num++;
		}
		lie2--;
		
	
		for(int i=lie2;i>=lie1&&num<N;i--)
		{
			matrix[hang2][i]=v[num];
			num++;
		}
	    hang2--;
		
		
		for(int i=hang2;i>=hang1&&num<N;i--)
		{
			matrix[i][lie1]=v[num];
			num++;
		}
	    lie1++;
		} 
	
	for(int i=0;i<m;i++)
	for(int j=0;j<n;j++)
	{
		cout<<matrix[i][j];
		if(j==n-1)
		cout<<endl;
		else cout<<" ";
	}
	return 0;
 }