[NOIP2014 普及组] 螺旋矩阵

题目背景

NOIP2014 普及组 T3

题目描述

一个 nn 行 nn 列的螺旋矩阵可由如下方法生成：

从矩阵的左上角（第 11 行第 11 列）出发，初始时向右移动；如果前方是未曾经过的格子，则继续前进，否则右转；重复上述操作直至经过矩阵中所有格子。根据经过顺序，在格子中依次填入 1, 2, 3, ... , n^21,2,3,...,n2，便构成了一个螺旋矩阵。

下图是一个 n = 4n=4 时的螺旋矩阵。

 1     2     3     4
12    13    14     5
11    16    15     6
10     9     8     7

现给出矩阵大小 nn 以及 ii 和 jj，请你求出该矩阵中第 ii 行第 jj 列的数是多少。

输入格式

共一行，包含三个整数 nn, ii, jj，每两个整数之间用一个空格隔开，分别表示矩阵大小、待求的数所在的行号和列号。

输出格式

一个整数，表示相应矩阵中第 ii 行第 jj 列的数。

输入输出样例

输入 #1复制

4 2 3

输出 #1复制

14

说明/提示

【数据说明】

对于 50\%50% 的数据，1 \leqslant n \leqslant 1001⩽n⩽100;

对于 100\%100% 的数据，1 \leqslant n \leqslant 30,000,1 \leqslant i \leqslant n,1 \leqslant j \leqslant n1⩽n⩽30,000,1⩽i⩽n,1⩽j⩽n。

第一次提交，我打暴力，先构建出一个螺旋矩阵，然后找出指定位置的值。代码大概是这样的：

#include <iostream>

int main() {
    int n, x, y;
    std::cin >> n >> x >> y;
    int a[n][n];
    int tmp = 1;
    for (int i = 0; i < n / 2 + 1; ++i) {
        for(int j = i; j < n - i; ++j)
        	a[i][j]=tmp++;
        for(int j = i + 1; j < n - i; ++j)
        	a[j][n-i-1]=tmp++;
        for(int j = n - i - 2; j > i; --j)
        	a[n-i-1][j]=tmp++;
        for(int j = n - i - 1; j > i; --j)
        	a[j][i]=tmp++;
    }
    std::cout << a[x-1][y-1] << std::endl;
}

但是，只有 5050 分，剩下的全都 TLE 了。

由于我太蒻了，想不出来如何在暴力算法上优化，所以换了一种方法——画出矩阵、观察规律、推导公式。

首先，我们画一个 5 \times 55×5 的螺旋矩阵。

观察一下规律，跟着数字增长的方向走，不难发现：

1. 如果是第 11 行，那么第 jj 列的数字就是 jj；

2. 如果是第 nn 列，那么第 ii 行的数字就是 n + i - 1n+i−1；

后两条规律有点难找，但是不要放弃，继续观察：

1. 如果是第 nn 行，那么第 jj 列的数字就是 3 \times n - 2 - j + 13×n−2−j+1；

2. 如果是第 11 列，那么第 ii 行的数字就是 4 \times n - 4 - i + 24×n−4−i+2。

好，现在对于每一种情况，我们都推出了一个公式。现在画一个 6 × 66×6 的螺旋矩阵，验证一下，会发现完全正确：

如果对于上述推导过程不是很理解，不妨打开 Excel，自己画图观察。

推导完公式，剩下的就简单多了。不难想出一个递归解法：把螺旋矩阵一层一层地剖开，看看目标位置在哪一层，然后加上这一层最左上角的数字（4 \times (n - 1)4×(n−1)），即为要求的数字。

于是，递归函数就可以写出来了：

int work(int n, int i, int j) {
    if (i == 1)
    	return j;
    if (j == n)
    	return n + i - 1;
    if (i == n)
    	return 3 * n - 2 - j + 1;
    if (j == 1)
    	return 4 * n - 4 - i + 2;
    // 注意，递归的时候，n 要减 2 而不是减 1
    return work(n - 2, i - 1, j - 1) + 4 * (n - 1);
}