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>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <cstring>

using namespace std;

vector<vector<int>> matrix;

int dx[] = {0, 1, 0, -1};
int dy[] = {1, 0, -1, 0};
void fill(const vector<int> &v, int row, int col)
{
    int index = 0, x = 0, y = 0, dir = 0;
    while(index < v.size())
    {
        matrix[x][y] = v[index++];
        if(dir == 0 && (y == col - 1 || matrix[x][y+1] != 0)) dir = 1;
        else if(dir == 1 && (x == row - 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];
    }
}

void findMN(int N, int &m, int &n)
{
    m = sqrt(N);
    while(N % m != 0) m++;
    n = N / m;
    if(m < n) swap(m, n);
}



int main()
{
    int N;
    cin >> 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, n;
    findMN(N, m, n);
    matrix.resize(m);
    for (int i = 0; i < m; ++i)
    {
        matrix[i].resize(n);
    }
    fill(v, m, n);
    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");
        }
    }
    return 0;
}