PAT 1050 螺旋矩阵

1050 螺旋矩阵（25 分）
本题要求将给定的 N 个正整数按非递增的顺序，填入“螺旋矩阵”。所谓“螺旋矩阵”，是指从左上角第 1 个格子开始，按顺时针螺旋方向填充。要求矩阵的规模为 m 行 n 列，满足条件：m×n 等于 N；m≥n；且 m−n 取所有可能值中的最小值。
输入格式：
输入在第 1 行中给出一个正整数 N，第 2 行给出 N 个待填充的正整数。所有数字不超过 $10 ^4$，相邻数字以空格分隔。
输出格式：
输出螺旋矩阵。每行 n 个数字，共 m 行。相邻数字以 1 个空格分隔，行末不得有多余空格。
输入样例：

12
37 76 20 98 76 42 53 95 60 81 58 93

输出样例：

98 95 93
42 37 81
53 20 76
58 60 76

解析

题目要求的螺旋矩阵就是上图所示，只不过是数字是从大到小。
这题我最后一个测试点没过，看了很多blog，哎，这题算了。≡(▔﹏▔)≡。心累
code:

#include<bits/stdc++.h>
using namespace std;
int getcolumn(int N){
    int sqr = sqrt(1.0*N);  
    for(;sqr!=0;sqr--)
        if(N%sqr==0)
            break;
    return sqr;
}
bool cmp(int a,int b){  return a>b;}
int data[10010];
int matrix[10010][10010];
int main()
{
    int N,m,n;
    scanf("%d",&N);
    for(int i=0;i<N;i++)
        scanf("%d",data+i);
    n = getcolumn(N);
    m = N/n;
    sort(data, data+N,cmp);
    int i=0,j=0,k=0,i_board=0,j_board=0,real_n=n,real_m=m;  
    while(k<N){
        while(j<n && k<N)
            matrix[i][j++] = data[k++];
        j--,i++;
        while(i<m && k<N)
            matrix[i++][j] =data[k++];
        j--,i--;        
        while(j>=j_board && k<N)
            matrix[i][j--] = data[k++];
        j++,i--;
        while(i>i_board && k<N)
            matrix[i--][j]  = data[k++];
        j++,i++,n--,m--,j_board++,i_board++;
    }
    for(int i=0;i<real_m;i++){
        for(int j=0;j<real_n;j++)
            printf("%d%c",matrix[i][j],j==real_n-1?'\n':' ');
    }
    return 0;
}