PAT——1105 Spiral Matrix 甲级

题目

https://pintia.cn/problem-sets/994805342720868352/problems/994805363117768704

题意

给出N个数字，按照从大到小的顺序螺旋输出m×n 的矩阵。其中m×n=N，m>=n且m与n的差值必须最小

代码解析

首先在sqrt（N）的左侧寻找符合要求的n，并根据n求出m

将得到的数组排序后，开始螺旋填充数组中的数字。

螺旋可以理解为遇到一面墙就转弯，如果没有遇到墙就直行。而在转弯后，墙要向中心压缩一格（因为原来墙的位置已经填充过了）

图中黑色箭头为数字行进方向，绿色箭头为墙壁行进方向

AC代码

#include<bits/stdc++.h>
using namespace std;
int main()
{
	int N,m,n;
	cin>>N;
	int a[N];
	for(int i=sqrt(N);i>=1;i--)
	{
		if(N%i==0)
		{
			n=i;
			m=N/n;
			break;
		}
	}
	int ans[m][n];
	for(int i=0;i<N;i++)
		cin>>a[i];
	sort(a,a+N,greater<int>());
	int flag=1,wall1=n-1,wall2=m-1,wall3=0,wall4=1,i=0,j=0,cnt=0;
	while(cnt<N)
	{
		if(flag==1)
		{
			ans[i][j++]=a[cnt++];
			if(j==wall1) 
			{
				flag=2;wall1--;
			}
		}
		else if(flag==2)
		{
			ans[i++][j]=a[cnt++];
			if(i==wall2) 
			{
				flag=3;wall2--;
			}
		}
		else if(flag==3)
		{
			ans[i][j--]=a[cnt++];
			if(j==wall3) 
			{
				flag=4;wall3++;
			}
		}
		else if(flag==4)
		{
			ans[i--][j]=a[cnt++];
			if(i==wall4) 
			{
				flag=1;wall4++;	
			} 
		}
	}
	for(i=0;i<m;i++)
		for(j=0;j<n;j++)
			printf("%d%s",ans[i][j],j<n-1?" ":"\n");
} 

参考

天梯赛题目练习——螺旋方阵（附带PTA测试点）