PAT A1105 Spiral Matrix (25 分) 模拟

    题目大意：给出N个数字，要求将这些数按照从大到小的顺序螺旋填入一个 m * n 的矩阵中，m > n 且 m - n取最小值。

    核心问题是如何将数字螺旋填入矩阵，如果定义要填入位置的坐标是(x , y) ，初始值显然是(0, 0)， 要在每次填入之后决定下一次填入的位置坐标。坐标轴取列增大的方向为y的正方向， 行增大的方向为 x的正方向， 然后定义两个增量数组 dx[ ] = {0, 1, 0, -1}， dy[ ] = {1, 0, -1, 0}， 那么就依次对应了 ①向右 ②向下 ③向左 ④向上， 下一次填入的位置坐标就是 （x + dx[ i ], y + dy[ i ] )。 最后的问题就是控制装填的方向，因为题目中说了填入的数字都是正数， 那么判断是否要转向的条件就是 （当前位置到达了矩阵的边界 || 按照当前方向的下一个位置已经不是0），然后变换方向即可。

AC代码：

#include <iostream>
#include <vector>
#include <algorithm>
#include <cstdio>
#include <cmath>

using namespace std;

void fillMatrix(int m, int n, vector<int> &v)
{
    vector<vector<int>> matrix(m);
    for (int i = 0; i < m; ++i)
        matrix[i].resize(n);
    int dx[] = {0, 1, 0, -1};
    int dy[] = {1, 0, -1, 0};
    int cnt = 0, x = 0, y = 0, dir = 0;
    while(cnt < v.size())
    {
        matrix[x][y] = v[cnt++];
        if(dir == 0 && (y == n-1 || matrix[x][y+1] > 0)) dir = 1;
        else if(dir == 1 &&(x == m - 1 || matrix[x+1][y] > 0)) dir = 2;
        else if(dir == 2 && (y == 0 || matrix[x][y-1] > 0)) dir = 3;
        else if(dir == 3 && (x == 0 || matrix[x-1][y] > 0)) dir = 0;
        x += dx[dir];
        y += dy[dir];
    }
    for (int i = 0; i < m; ++i)
    {
        for (int j = 0; j < n; ++j)
        {
            printf("%d", matrix[i][j]);
            if(j < n-1) printf(" ");
            else printf("\n");
        }
    }
}

int main()
{
    int N;
    scanf("%d", &N);
    vector<int> v(N);
    for (int i = 0; i < N; ++i)
        scanf("%d", &v[i]);
    sort(v.begin(), v.end(), greater<int>());
    int m = 1, n = 1;
    for (m = sqrt(N) + 1; m >= 1; --m)
    {
        if(N % m == 0)
        {
            n = N / m;
            break;
        }
    }
    if(m < n) swap(m, n);
    fillMatrix(m, n, v);
    return 0;
}