PAT (Advanced Level) 1105. Spiral Matrix (25)

#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include<queue>
#include<string>
#include<algorithm>
using namespace std;

const int maxn=10000+10;
int N,a[maxn];
int ans[maxn];
int m,n;
int dir[4][2];

void f()
{
dir[0][0]=0;dir[0][1]=1;
dir[1][0]=1;dir[1][1]=0;
dir[2][0]=0;dir[2][1]=-1;
dir[3][0]=-1;dir[3][1]=0;
}

bool cmp(const int&a,const int&b)
{
return a>b;
}

int main()
{
scanf("%d",&N);
f();
for(int i=1;i<=N;i++) scanf("%d",&a[i]);
sort(a+1,a+1+N,cmp);

for(int i=sqrt(1.0*N);i>=1;i--)
if(N%i==0){n=i,m=N/i;break;}

memset(ans,-1,sizeof ans);
for(int i=1;i<=m;i++)
for(int j=1;j<=n;j++)
ans[(i-1)*n+j]=0;

int nowDir=0;
int nowX=1,nowY=1;
for(int i=1;i<=N;i++)
{
ans[(nowX-1)*n+nowY]=a[i];
int NewX=nowX+dir[nowDir][0];
int NewY=nowY+dir[nowDir][1];
if(ans[(NewX-1)*n+NewY]!=0||NewX<=0||NewX>=m+1||NewY<=0||NewY>=n+1) nowDir=(nowDir+1)%4;
nowX=nowX+dir[nowDir][0];
nowY=nowY+dir[nowDir][1];
}
int num=1;
// for(int i=1;i<=m*n;i++) printf("%d ",ans[i]);

for(int i=1;i<=m;i++)
for(int j=1;j<=n;j++)
{
printf("%d",ans[num++]);
if(j<n) printf(" ");
else printf("\n");
}
return 0;
}