牛客16502 螺旋矩阵

链接：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 j
    if i == n:
        return 3*n-2-j+1
    if j == 1:
        return 4*n-2-i
    if j == n:
        return n+i-1
    return fn(n-2, i-1, j-1)+4*n-4

print(fn(n, i, j))

这里有个问题

Python中默认的最大递归深度是989，当尝试递归第990时便出现递归深度超限的错误：

RuntimeError: maximum recursion depth exceeded in comparison

可以手动设置递归调用深度：

import sys
sys.setrecursionlimit(10000000)