PAT 乙级 1050 螺旋矩阵 (25 分)

---

题目

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

---

分析

一、获取N，并根据N计算m、n
将N开方后的整数部分赋值给n,为了保证保证m*n=N，需要n能整除N；因此令n–直至可以整除，c此时m=N\n。同时该做法保证了m-n的差最小。

二、获取N个正整数并降序排序

三、仿真构建二维数组b[m][n]
1.将二维数组的值初始化为-1，用于标志其对应位置尚未赋值。
2.首先令i=0，j=0,即将当前需要输入的位置指向二维矩阵的最上角
3.在构建螺旋矩阵时，共存在4种情况：
    ①从左到右：j++
    ②从上到下：i++
    ③从右到左：j–
    ④从下到上：i–
用flag%4标记以上四种情况
4.边界条件判断
在每次输入前需要判断当前输出的位置是否正确，以下两种情况表示非法，需要变换方向：
    ①b[i][j]>0 此时表示该位置已经输入过
    ②i=m或j=n 此时需要输入的位置已经超出了二维数组的范围
此时位置非法，需要通过改变i j的值来改变当前输入位置，同时变更输入方向(令flag++),上述四种情况分别对应以下处理方法：
    ①从左到右：j-1,i+1;flag++
    ②从上到下：i-1,j-1;flag++
    ③从右到左：j+1,i-1;flag++
    ④从下到上：i+1,j+1;flag++

四、按行列顺序输出二维数组

---

代码

#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
using namespace std;
bool cmp(int x,int y){return x>y;}
int main(){
    int i,j,N,m,n,num,flag=0;
    scanf("%d",&N);
    for (n = sqrt((double)N); n >= 1; n--) {
        if (N % n == 0) {
            m = N / n;
            break;
        }
    }
    int a[N],b[m][n];
    memset(b, -1, sizeof(b)); 
    for(i = 0; i < N; i++) scanf("%d",&a[i]);
    sort(a,a+N,cmp);
    i=0;j=0;
    for(num = 0; num < N; num++){
        switch(flag%4){
            case 0:
                    if(j < n && b[i][j] < 0){
                        b[i][j++]=a[num];
                    }else{
                        flag++;
                        b[(++i)++][--j]=a[num];
                    }break;
            case 1:
                    if(i < m && b[i][j] < 0){
                        b[i++][j]=a[num];
                    }else{
                        flag++;
                        b[--i][(--j)--]=a[num];
                    }break;
             case 2:
                    if(j >= 0 && b[i][j] < 0){
                        b[i][j--]=a[num];
                    }else{
                        flag++;
                        b[(--i)--][++j]=a[num];
                    }break;
              case 3:
                    if(i >= 0 && b[i][j] < 0){
                        b[i--][j]=a[num];
                    }else{
                        flag++;
                        b[++i][(++j)++]=a[num];
                    }break;
        }
    }
    for(i=0;i<m;i++){
        for(j=0;j<n;j++){
            if(j>0) printf(" ");
            printf("%d",b[i][j]);
        }
        printf("\n");
    }
    return 0;
}

---

总结

二维数组的初始化好像不能像一维数组一样直接b[m][n]={-1};然后就用了memset()这个C语言string库里的函数，这个函数初始化的值只能为0或-1。

下面这种方法对于初始化同一个值也挺好用

#define ARR_LEN 100
int arr4[ARR_LEN][ARR_LEN] = { [0 ... (ARR_LEN-1)][0 ... (ARR_LEN-1)] = 10 }; /* 100*100个元素都初始化为10 */

做题的时候一开始草率了，就直接n=floor(sqrt(N));m=N\n;忘记考虑N%n!=0的情况了导致最开始只拿了20分，改完以后就AC了。