PAT乙级1050 螺旋矩阵 (25 分)

Posted 2021-08-08 辉小歌

题目地址

#include<cstdio>
#include<iostream>
#include<string>
#include<vector>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=10005;

int a[10005][105];
int s[N];
int k,n,m;

int dx[4]={0,1,0,-1};
int dy[4]={1,0,-1,0};
bool cmp(int x,int y)
{
	return x>y;
}
int main(void)
{
	cin>>k;
	for(int i=0;i<k;i++) cin>>s[i];
	if(k==0) return 0;
	for(int i=1;i<=sqrt(k);i++) if(k%i==0) n=i,m=k/i;
	
	sort(s,s+k,cmp);
	memset(a,0x3f,sizeof a);
	
	int x=0,y=0,d=0;
	int index=0;
	while(index<k)
	{
		a[x][y]=s[index++];
		int tempx=x+dx[d];
		int tempy=y+dy[d];
		if(tempx<0||tempx>=m||tempy<0||tempy>=n||a[tempx][tempy]!=0x3f3f3f3f) d=(d+1)%4;	
		x=x+dx[d];
		y=y+dy[d];
	}
	for(int i=0;i<m;i++)
	{
		for(int j=0;j<n;j++)
		{
			cout<<a[i][j];
            if(j+1!=n) cout<<" ";
		}
		cout<<endl;
	}
	return 0;
}