链接:https://ac.nowcoder.com/acm/problem/16502
来源:牛客网
题目描述
一个 n 行 n 列的螺旋矩阵可由如下方法生成:
从矩阵的左上角(第 1 行第 1 列)出发,初始时向右移动;如果前方是未曾经过的格子,则继续前进,否则右转;重复上述操作直至经过矩阵中所有格子。根据经过顺序,在格子中依次填入 1, 2, 3, … , n ,便构成了一个螺旋矩阵。
下图是一个 n = 4 时的螺旋矩阵。
现给出矩阵大小 n 以及 i 和 j ,请你求出该矩阵中第 i 行第 j 列的数是多少。
输入描述:
输入共一行,包含三个整数 n,i,j ,每两个整数之间用一个空格隔开,分别表示矩阵大小、待求的数所在的行号和列号。
输出描述:
输出一个整数,表示相应矩阵中第 i 行第 j 列的数。
找规律:
当 i == 1
时,res = j
当 i == n
时,res = 3n-2-j+1
当 j == 1
时,res = 4n-2-i
当 j == n
时, res = n+i-1
然后递归, 每减少一层 ==> n-2, i-1, j-1,并且要加上外层的 4n-4
import sys
sys.setrecursionlimit(10000000)
n, i, j = map(int,input().split())def fn(n, i, j):if i == 1:return jif i == n:return 3*n-2-j+1if j == 1:return 4*n-2-iif j == n:return n+i-1return fn(n-2, i-1, j-1)+4*n-4print(fn(n, i, j))
这里有个问题
Python中默认的最大递归深度是989,当尝试递归第990时便出现递归深度超限的错误:
RuntimeError: maximum recursion depth exceeded in comparison
可以手动设置递归调用深度:
import sys
sys.setrecursionlimit(10000000)