2014NOIP普及组真题 3. 螺旋矩阵

线上OJ：

一本通：http://ybt.ssoier.cn:8088/problem_show.php?pid=1967

背景知识：

螺旋矩阵可以采用模拟的方式生成。就是顺时针四个方向

第1步、是第 1 行，方向为从左到右，数值+1。当向右遇到 边界n 或者 格子已填过数值 时停止
第2步、是第 n 列，方向为从上到下，数值+1。当向下遇到 边界n 或者 格子已填过数值 时停止
第3步、是第 n 行，方向为从右到左，数值+1。当向左遇到 边界0 或者 格子已填过数值 时停止
第4步、是第 1 列，方向为从下到上，数值+1。当向上遇到 边界0 或者 格子已填过数值 时停止

以上四步为一轮。然后开始下一轮，直至数值填满达到 n*n 为止。

本题的第一种常规解法，是先生成螺旋矩阵，然后输出其中的数值。但是题中的n会达到30000，此时开a[n][n]的二位矩阵在空间上会爆，所以这种方法只能拿50分。

代码如下：

解法一（50%分数）

#include <bits/stdc++.h>
using namespace std;
const int maxn = 10005;
int a[maxn][maxn];
int n, i ,j;
int dx[4]={0, 1, 0, -1};
int dy[4]={1, 0, -1, 0};

void cal() {
    int x = 0, y = 0;
    a[x][y] = 1;
    for (int k = 2; k <= n * n; )
    {
        for(int t = 0; t < 4; t++)
        {
            while (1)
            {
                int nx = x + dx[t], ny = y + dy[t];
                if (nx < 0 || nx >= n || ny < 0 || ny >= n || a[nx][ny])  break;

                a[x = nx][y = ny] = k++;
            }
        }     
    }
}
int main() {
    cin >> n >> i >> j;
    cal();
    cout << a[i-1][j-1];
    return 0;
}

以上代码的二位矩阵会爆，故需要进行优化。由于本题中只需要输出 a[i][j] 的值，并不需要生成整个螺旋矩阵。故可以考虑通过递归来计算 a[i][j]。

核心思想：

由于螺旋矩阵每次只能顺序计算最外圈，故我们可以根据矩阵的 n 和 i j 推断出最外圈每个位置的计算公式（如下图所示）。

公式1（外圈数值）：

第 1 行，$a[1][j]=j$ // 如果剥离外层后，所求点处于当前矩阵的第一行，则行坐标就是值
第 n 列，$a[i][n]=n+i-1$ // 如果剥离外层后，所求点处于当前矩阵的最后一列，则 i+n-1 就是值（从上往下走）
第 n 行，$a[n][j]=3n-j-1$ // 如果剥离外层后，所求点处于当前矩阵的最后一行，则 2n-1+n-j = 3n-j-1 就是值
第 1 列，$a[i][1]=4n-i-2$ // 如果剥离外层后，所求点处于当前矩阵的第一列，则 3n-2+n-i= 4n-i-2就是值

光有外圈的计算公式还不足以计算内圈a[i][j]，需要知晓每次剥去外圈向内传递时的差值。

从上图中，我们可以发现，a[1][1]=1，传到a[2][2]的增量为当前 n-1 的4倍。

比如：n=9 时，增量为4*(9-1)=32
比如：n=7 时，增量为4*(7-1)=24
比如：n=5 时，增量为4*(5-1)=16

综上所述，可以先深搜 a[i][j] 达到最外层，使其可直接计算。然后再层层递归回来，每次**递归回时加上增量 4(n-1) **即可。
代码如下

解法二、dfs（100%）

#include <bits/stdc++.h>
using namespace std;

int n, i, j;

// 剥离外圈向内传递时，n减少2（因为行、列各减少2），i和j各减少1
// a[1][1]到a[2][2]的增量为 (n-1)*4
int dfs(int n, int i, int j)
{
    if(i == 1)  return j;             // 如果剥离外层后，所求点处于当前矩阵的第一行，则行坐标就是值
    if(j == n)  return n + i - 1;     // 如果剥离外层后，所求点处于当前矩阵的最后一列，则 i+n-1 就是值（从上往下走）
    if(i == n)  return 3 * n - j - 1; // 如果剥离外层后，所求点处于当前矩阵的最后一行，则 2n-1+n-j = 3n-j-1 就是值
    if(j == 1)  return 4 * n - i - 2; // 如果剥离外层后，所求点处于当前矩阵的第一列，则 3n-2+n-i= 4n-i-2就是值

    return dfs(n-2, i-1, j-1) + (n-1)*4;  // 如果剥离外层后，所求点还没到外圈，则继续剥离本圈
}

// 在已知 n 的前提下，最外圈的坐标可以直接求出
// 如果不在最外圈，就一层一层剥离，直到所求点处于最外圈
int main()
{
    cin >> n >> i >> j;
    cout << dfs(n, i, j);
    return 0;
}

在解法二中采用了深搜，当 a[i][j] 处于内圈时，依然有较高的复杂度。实际上，当我们在分析图二时，已经可以发现，a[1][1]可以一直向下传递。

其中，
1、a[1][1] → a[2][2]，增量为 32，即 4*(9-1) = 4*(n-1)
2、a[2][2] → a[3][3]，增量为 24，即 4*(9-1) - 8 = 4*(n-1) - 8。所以 $$a[3][3]=a[2][2]+4\ast (n-1)-8=2\ast 4\ast (n-1)-8$$
3、a[3][3] → a[4][4]，增量为 16，即 4*(9-1) - 8 - 8 = 4*(n-1) - 28。所以 $$a[4][4]=a[3][3]+4\ast (n-1)-2\ast 8=3\ast 4\ast (n-1)-3\ast 8$$
4、a[4][4] → a[5][5]，增量为 8，即 4(9-1) - 8 - 8-8 = 4*(n-1) - 3*8。所以 $$a[5][5]=a[4][4]+4\ast (n-1)-3\ast 8=4\ast 4\ast (n-1)-6\ast 8$$
所以，剥离 k 层后的起始元素可表示为： $k\ast 4\ast (n-1)-(k-1)\ast (8+(k-1)\ast 8)/2$
也就是说，我们不需要通过 dfs 也可以直接穿透到 a[i][j] 所在的那一个圈。

公式2（剥离 k 层后的起始元素）：
$st=k\ast 4\ast (n-1)-(k-1)\ast (8+(k-1)\ast 8)/2$

步骤如下（假设我们要计算是的 a[4][5]）：

第一步：找到 a[4][5] 所在圈的第一个数值 a[4][4]， 从a[1][1]可以直接推出
$a[4][4]=k\ast 4\ast (n-1)-(k-1)\ast (8+(k-1)\ast 8)/2$
第二步：在a[4][5] 所在圈内，按照公式1推出增量
第三步：前两步的数值相加

公式推导后的时间复杂度为O(1)，代码见下

解法三、公式推导 O(1)

#include <bits/stdc++.h>
using namespace std;

int n, i, j;

// 剥离外圈向下传递时，n减少2（因为行、列各减少2），ℹ和j各减少1
int solve(int n, int i, int j)
{
    if(i == 1)  return j;             // 如果剥离外层后，所求点处于当前矩阵的第一行，则行坐标就是值
    if(j == n)  return n + i - 1;     // 如果剥离外层后，所求点处于当前矩阵的最后一列，则 i+n-1 就是值（从上往下走）
    if(i == n)  return 3 * n - j - 1; // 如果剥离外层后，所求点处于当前矩阵的最后一行，则 2n-1+n-j = 3n-j-1 就是值
    if(j == 1)  return 4 * n - i - 2; // 如果剥离外层后，所求点处于当前矩阵的第一列，则 3n-2+n-i= 4n-i-2就是值
}

// a[i][j] = st + increase
int main()
{
    cin >> n >> i >> j;
    if(i > n || j > n)
        return 0;
    else if(i == 1 || j == 1)
        cout << solve(n, i, j);
    else
    {
        int k = min(min(n-i, n-j), min(i, j) - 1); // a[i][j]处于第k层
        // 4(n-1)-0*8+4(n-1)-1*8+4(n-1)-2*8+4(n-1)-3*8 = k*4*(n-1) - (0*8+1*8+2*8+3*8+...(k-1)*8) = (k-1)(8+(k-1)*8)/2
        int st = k*4*(n-1) - (k-1)*(8+(k-1)*8)/2;

        cout << st + solve(n-2*k, i-k, j-k);
    }

    return 0;
}