PAT basic-level 1050 螺旋矩阵 笔记

1050. 螺旋矩阵(25)

时间限制150 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
CHEN, Yue
本题要求将给定的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

---

这个题有点迷宫问题的意思，难点就两个，一个算出m和n，难度不大，一个是螺旋填充，要考虑清楚，思路是：
用四个变量isEast、isSouth…来记录当前行径方向，通过判断前进方向是否满足条件来转变方向。要注意的是，在判断是否可以按当前前进方向继续前进的时候，要预先往前进一位，看看这个位是否超出或者已被填充，那么就要求进位之后不能超出数二维数组范围，于是我是建立一个(m+2) * (n+2)的二维数组，填充的是中间m*n的部分，这样就不会出现越界情况。

#include <iostream>
#include <vector>
#include <cmath>
#include <algorithm>
using namespace std;

bool cmp(int a, int b){
    return a > b;
}

int main(){
    int num;
    cin >> num;
    int m, n;
    double d = sqrt(num);
    if ((d - (int)d) != 0){
        for (int i = 1; i <= (int)d; i++){
            if ((num % i) != 0)
                continue;
            else {
                n = i;
                m = num / n;
            }
        }
    } else m = n = (int)d;
    vector<int> arr(num, 0);
    for (int i = 0; i < num; i++)
        cin >> arr[i];
    sort(arr.begin(), arr.end(), cmp);
    vector< vector<int> > show(m+2, vector<int>(n+2, 0));   //多两位，防止访问非法

    int ii = 1, jj = 1; //ii行，jj列
    int isEast = 1, isSouth = 0, isWest = 0, isNorth = 0;
    for (int k = 1; k <= num; k++){
        show[ii][jj] = arr[k - 1];

        if (isEast && (jj == n || show[ii][jj + 1] != 0)){
            isEast = 0;
            isSouth = 1;
        } else if (isSouth && (ii == m || show[ii + 1][jj] != 0)){
            isSouth = 0;
            isWest = 1;
        } else if (isWest && (jj == 1 || show[ii][jj - 1] != 0)){
            isWest = 0;
            isNorth = 1;
        } else if (isNorth && (ii == 1 || show[ii - 1][jj] != 0)){
            isNorth = 0;
            isEast = 1;
        }

        if (isEast) jj++;
        else if (isSouth)   ii++;
        else if (isWest) jj--;
        else if (isNorth) ii--;
    }
    for (int i = 1; i <= m; i++){
        for (int j = 1; j <= n; j++){
            cout << show[i][j];
            if (j != n)
                cout << ' ';
        }
        cout << endl;
    }
    return 0;
}