【PAT甲级】1105 Spiral Matrix (25分)

题意：
输入一个正整数N（实则<=1e5），接着输入一行N个正整数（<=1e4）。降序输出螺旋矩阵。

trick：

测试点1，3运行超时原因：直接用sqrt(N)来表示矩阵的宽会在N是素数时出错，直接扫一遍找一个最大因子即可，简单粗暴。

AAAAAccepted code：

#define HAVE_STRUCT_TIMESPEC
#include<bits/stdc++.h>
using namespace std;
int a[100007];
int ans[1007][1007];
int main(){
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    int n;
    cin>>n;
    for(int i=1;i<=n;++i)
        cin>>a[i];
    int y=0;
    for(int i=1;i*i<=n;++i)
        if(n%i==0)
            y=i;
    int x=n/y;
    sort(a+1,a+1+n);
    int top=1,bottom=x,left=1,right=y;
    int temp=n;
    int flag=1;
    while(temp){
        if(flag==1){
            for(int i=left;i<=right;++i)
                ans[top][i]=a[temp--];
            ++top;
            flag=2;
        }
        else if(flag==2){
            for(int i=top;i<=bottom;++i)
                ans[i][right]=a[temp--];
            --right;
            flag=3;
        }
        else if(flag==3){
            for(int i=right;i>=left;--i)
                ans[bottom][i]=a[temp--];
            --bottom;
            flag=4;
        }
        else if(flag==4){
            for(int i=bottom;i>=top;--i)
                ans[i][left]=a[temp--];
            ++left;
            flag=1;
        }
    }
    for(int i=1;i<=x;++i){
        for(int j=1;j<=y;++j){
            cout<<ans[i][j];
            if(j<y)
                cout<<" ";
        }
        if(i<x)
            cout<<"
";
    }
    return 0;
}