题目背景
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 的螺旋矩阵。
观察一下规律,跟着数字增长的方向走,不难发现:
-
如果是第 11 行,那么第 jj 列的数字就是 jj;
-
如果是第 nn 列,那么第 ii 行的数字就是 n + i - 1n+i−1;
后两条规律有点难找,但是不要放弃,继续观察:
-
如果是第 nn 行,那么第 jj 列的数字就是 3 \times n - 2 - j + 13×n−2−j+1;
-
如果是第 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);
}