1050. 螺旋矩阵(25)

题目

本题要求将给定的N个正整数按非递增的顺序，填入“螺旋矩阵”。所谓“螺旋矩阵”，是指从左上角第1个格子开始，按顺时针螺旋方向填充。要求矩阵的规模为m行n列，满足条件：m*n等于N；m>=n；且m-n取所有可能值中的最小值。

输入格式：

输入在第1行中给出一个正整数N，第2行给出N个待填充的正整数。所有数字不超过104，相邻数字以空格分隔。

输出格式：

输出螺旋矩阵。每行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

分析

思路：

1. 动态分配数组空间，并初始化（calloc），这个用熟练，可以避免一些数组访问越界的问题。
2. qsort的cmp的覆写
3. 剩下的就是分治的思想，将矩阵的输出简化为由外向内的圈的输出。

代码

#include<stdio.h>
#include<stdlib.h>
#include<algorithm>
#include<math.h>

int cmp(const void *a,const void *b){
    return *(int*)b - *(int*)a;  //注意这是非升序的写法（降序）
}
int main(){
    int N;
    scanf("%d",&N);
    int *a=(int*)calloc(N,sizeof(int));
    int *mat=(int*)calloc(N,sizeof(int));
    for(int i=0;i<N;i++){
        scanf("%d",a+i);
    }
    qsort(a,N,sizeof(int),cmp);
    int i=sqrt(N);
    while(!(i*i>=N&&N%i==0)) i++;
    int m=i,n=N/i;
    int y=0,x=-1;  //用来标志当前坐标
    int v=m,h=n;  //代表垂直和水平方向需要前进的步数
    int count=0;
    while(h>0&&v>0){
        for(int i=0;i<h&&v>0;i++) mat[y*n+ ++x]=a[count++];  //向右
        v--;  //垂直方向减一行
        for(int i=0;i<v&&h>0;i++) mat[++y*n+x]=a[count++];  //向下
        h--;  //水平方向减一行
        for(int i=0;i<h&&v>0;i++) mat[y*n + --x]=a[count++];  //向左
        v--;
        for(int i=0;i<v&&h>0;i++) mat[--y*n+x]=a[count++];  //向上
        h--;
    }
    for(int i=0;i<m;i++)
        for(int j=0;j<n;j++)  //对空格和换行的控制是经典写法
            printf("%d%c",mat[i*n+j],j==n-1 ? '\n' : ' ');
    return 0;
}