如何用递归实现Python最大平方金字塔数求解函数？

我需要定义一个函数，找到不大于参数n的最大平方金字塔数（平方金字塔数是1²、1²+2²、1²+2²+3²……这样的数）。我已经用迭代法成功实现了：

def largest_square_pyramidal_num(n):
    total = 0
    i = 0
    while total <= n:
        total += i**2
        i += 1
    if total > n:
        return total - (i-1)**2
    else:
        return total

但第二部分要求用递归实现时我卡壳了。我习惯的递归都是直接操作参数，但这里参数是判断条件，我尝试写了一段错误的代码：

def largest_square_pyramidal_num_rec(n):
    m = 0
    pyr_number = 0
    pyr_number += m**2
    def pyr_num(m):
        if pyr_number >= n:
            return pyr_number
        else:
            return pyr_num(m+1)
    return pyr_number

我知道这段代码有问题：外部定义的m和pyr_number在递归里有问题，而且每次调用pyr_num时pyr_number都会回到0，但我不知道怎么修正。我的思路是重复累加平方数得到金字塔数作为递归过程，但不知道怎么把这个过程和参数n结合起来，也不清楚该如何正确维护递归的状态。

你的核心问题在于递归过程中没有正确维护状态——你尝试用外部变量保存累加的金字塔数，但递归函数每次调用时都会重新初始化这些变量，导致状态丢失。递归的关键是把需要跟踪的状态（当前累加的金字塔数、下一个要加的整数）作为参数传递给递归函数，而不是依赖外部变量。

方案1：使用辅助递归函数 #

我们可以写一个主函数，内部定义一个辅助递归函数，用来传递当前累加值和当前的整数i：

def largest_square_pyramidal_num_rec(n):
    def helper(current_sum, current_i):
        # 计算加上下一个平方数后的候选值
        next_pyramidal = current_sum + current_i**2
        # 如果候选值不超过n，继续递归累加下一个平方数
        if next_pyramidal <= n:
            return helper(next_pyramidal, current_i + 1)
        # 如果超过了n，当前的current_sum就是最大的符合条件的数
        else:
            return current_sum
    # 初始状态：当前累加和为0，从i=1开始（因为第一个平方数是1²）
    return helper(0, 1)

方案2：直接用默认参数简化调用 #

如果不想用辅助函数，也可以把状态参数设置为默认值，这样调用函数时只需要传入n即可，和迭代版的调用方式一致：

def largest_square_pyramidal_num_rec(n, current_sum=0, current_i=1):
    next_pyramidal = current_sum + current_i**2
    if next_pyramidal <= n:
        # 递归调用，更新累加和和当前整数
        return largest_square_pyramidal_num_rec(n, next_pyramidal, current_i + 1)
    else:
        return current_sum

为什么这两个方案能解决你的问题？ #

1. 状态传递而非外部变量：每一层递归都会把当前的累加和、下一个要加的整数作为参数传递，不会出现外部变量被重置或共享的问题。
2. 递归终止条件清晰：当加上下一个平方数后超过n时，就返回当前的累加和，这就是我们要找的最大金字塔数。
3. 逻辑和迭代版对应：迭代版是循环累加直到超过n，递归版则是每次判断是否能继续累加，能就递归，不能就返回，逻辑完全一致。

举个例子，当n=14时：

• 初始调用：helper(0,1) → 计算0+1=1 ≤14，递归helper(1,2)
• 计算1+4=5 ≤14，递归helper(5,3)
• 计算5+9=14 ≤14，递归helper(14,4)
• 计算14+16=30 >14，返回14，这就是正确结果。

内容的提问来源于stack exchange，提问作者a9302c