python - 找到n个有趣的多边形区域

Question

我试图找到一个 n-interesting 多边形的面积，其中 (n=1, A=1, n=2, A=5, n=3, A=13, n=4, A=25, 等等在)。所以n-interesting多边形的公式是(n-1)-interesting多边形的面积+(n-1)*4。运行程序时，隐藏测试显示代码错误。我无法弄清楚我的代码有什么问题。

def shapeArea(n):
    if n == 0:
        return 0
    if n == 1:
        return 1
    for i in range(2,n+1):
        return (shapeArea(n-1)+(n-1)*4)

Answer 1

我找到了没有递归的公式。测试顺利通过。

def shapeArea(n):
    if n>=10**4 or n<1:
        return False

    return (n**2+(n-1)**2)