【好题】【天梯赛】1050 螺旋矩阵 (25 分)

原题链接

我感觉这道题挺难的。。看的题解

#include<bits/stdc++.h>
using namespace std;
int k,m,n;
bool cmp(int x,int y)
{
	return x>y;
}
int main()
{
	scanf("%d",&k);
	vector<int> a(k);
	for(int i=0;i<k;i++)
	{
		scanf("%d",&a[i]);
	}
	for(int i=sqrt((double)k);i>=1;i--)
	{
		if(k%i==0)
		{
			m=k/i;
			n=i;//每行数量 
			break;
		}
	}
	sort(a.begin(),a.end(),cmp);
	int level=m/2+m%2,t=0;
	//这儿 实在不懂。。。。 
	vector< vector<int> >b(m,vector<int>(n) );
	for(int i=0; i<level; i++)
	{
		for(int j=i;j<=n-1-i&&t<=k-1;j++)
		{
			b[i][j]=a[t++];
		}
		for(int j=i+1;j<=m-2-i&&t<=k-1;j++)
		{
			b[j][n-1-i]=a[t++];
		}
		for(int j=n-1-i;j>=i&&t<=k-1;j--)
		{
			b[m-i-1][j]=a[t++];
		}
		for(int j=m-2-i;j>=i+1&&t<=k-1;j--)
		{
			b[j][i]=a[t++];
		}
	}
	for(int i=0;i<m;i++)
	{
		for(int j=0;j<n;j++)
		{
			printf("%d",b[i][j]);
			if(j!=n-1) cout<<" ";
		}
		cout<<endl;
	}
	return 0;
}