PAT(Basic Level)_1050_螺旋矩阵

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<functional>

using namespace std;


int main(){
    int N,m,n,nums[10000];
    scanf("%d",&N);
    for(int i=0;i<N;i++){
        scanf("%d",nums+i);
    }
    sort(nums,nums+N,greater<int>());
    //for(int i=0;i<N;i++) printf("%d ",nums[i]); 

    for(int i=sqrt(N);i>0;i--){
        if(N%i==0){
            n=i;
            m=N/n;
            break;
        }
    }
    //printf("%d %d",m,n);

    int **matrix=new int*[m];
    for(int i=0;i<m;i++){
        matrix[i]=new int[n];
    }
    for(int i=0;i<m;i++){
        for(int j=0;j<n;j++){
            matrix[i][j]=0;
        }
    }

    int i=0,dir=0,x=0,y=0;
    while(i<N){
        if(!matrix[x][y]) matrix[x][y]=nums[i++];
        switch(dir){
            case 0:
                if(y+1<n && !matrix[x][y+1]) y++;
                else dir=1;
                break;
            case 1:
                if(x+1<m && !matrix[x+1][y]) x++;
                else dir=2;
                break;
            case 2:
                if(y-1>=0 && !matrix[x][y-1]) y--;
                else dir=3;
                break;
            case 3:
                if(x-1>=0 && !matrix[x-1][y]) x--;
                else dir=0;
                break;
        }
    }

    for(int i=0;i<m;i++){
        for(int j=0;j<n;j++){
            printf("%d",matrix[i][j]);
            if(j==n-1) putchar('\n');
            else putchar(' ');
        }
    }

    for(int i=0;i<m;i++){
        delete[]matrix[i];
    }
    delete[]matrix;

    return 0;
}