如何修复质子穿铝能损计算中递归函数的栈溢出错误？

你的需求是计算质子穿过20000层0.025µm铝片后的剩余能量，每层需根据当前能量更新阻止本领并扣除对应能量消耗。但原递归实现即使调高递归限制仍出现栈溢出，核心原因是Python不支持尾递归优化，20000次递归调用超出了系统栈的实际承载上限。

解决方案：将递归改为迭代实现 #

迭代方式无需不断创建新的栈帧，完全避免栈溢出问题，同时运行效率更高。修改后的代码如下：

def calculate_remaining_energy(initial_E, initial_stop):
    step = 0
    current_E = initial_E
    current_stop_reciprocal = initial_stop  # 对应原代码中的n，是阻止本领的倒数
    while step < 20000:
        # 扣除当前层的能量：能量 = 当前能量 - 层厚度 * 阻止本领
        current_E = current_E - 0.025 * (1 / current_stop_reciprocal)
        # 计算新的阻止本领倒数
        current_stop_reciprocal = (-0.0079199 
                                  + 0.040703 * current_E**(-0.4) 
                                  + 0.0022677 * current_E**(0.25) 
                                  + 0.000058309 * current_E**(0.8))
        step += 1
    return current_E

# 初始参数调用
print(calculate_remaining_energy(13000, 0.131234))

关键说明 #

1. 递归的局限性：Python解释器默认不优化尾递归，即使通过sys.setrecursionlimit()调高数值，系统的物理栈大小依然有限，20000次递归调用会直接耗尽栈空间导致溢出。
2. 迭代的优势：迭代仅在单个栈帧内循环执行，通过变量跟踪当前步数、能量和阻止本领倒数，不会产生额外的栈开销，适合这种大规模的循环计算场景。

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