如何提升递归实现的阶乘计算程序运行效率？求优化建议

你编写的递归阶乘计算程序核心逻辑可行，但在鲁棒性、性能和用户体验上还有优化空间，以下是具体建议：

原代码 #

def factorial(base):
    if base == 0 or base == 1:      
        return 1                                  
    else:
        return base * factorial(base - 1)                            


while True:
   base: int = int(input("Enter a base number: "))
   Result = factorial(base)
   print(f"The factorial of {base} is: {Result}")
   offf: str = input("Enter 'off' to terminate calculations: ")
   if offf == "off":
    print("Calculations Terminated")
    break

程序终端输出展示了用户输入数字后获取阶乘结果，以及输入“off”终止程序的完整流程。

优化建议 #

• 添加输入合法性校验：当前代码直接将输入转为整数，若用户输入非数字、负数会触发报错。建议捕获ValueError异常，同时判断输入是否为非负整数，给出友好提示：

  try:
      base = int(input("输入一个非负整数（输入'off'终止程序）："))
      if base < 0:
          print("阶乘仅支持非负整数，请重新输入")
          continue
  except ValueError:
      user_input = input().strip()
      if user_input.lower() == 'off':
          break
      print("输入无效，请输入非负整数或'off'")
      continue
  

• 解决递归栈溢出问题：递归实现阶乘在计算大数（如超过1000）时会触发栈溢出错误。建议改用迭代实现，或直接使用Python标准库math.factorial（经过优化，能高效处理大数且内置边界校验）：

  import math
  
  # 替代自定义递归函数
  result = math.factorial(base)
  

  若坚持自定义，迭代实现示例：

  def factorial(base):
      if base <= 1:
          return 1
      result = 1
      for i in range(2, base + 1):
          result *= i
      return result
  

• 规范变量命名：遵循PEP8命名规范，变量名使用小写加下划线，比如将Result改为result，offf改为terminate_input，提升代码可读性。
• 优化终止逻辑：合并两次输入为一次，让用户可以直接在数字输入环节输入“off”终止，减少操作步骤，提升体验：

  while True:
      user_input = input("输入一个非负整数（输入'off'终止程序）：").strip()
      if user_input.lower() == 'off':
          print("计算已终止")
          break
      try:
          base = int(user_input)
          if base < 0:
              print("阶乘仅支持非负整数，请重新输入")
              continue
          result = factorial(base)
          print(f"{base}的阶乘是：{result}")
      except ValueError:
          print("输入无效，请输入非负整数或'off'")
  

• 简化递归函数逻辑：将递归的条件判断简化为if base <= 1:，无需分别判断0和1，逻辑更简洁。

优化后完整代码示例 #

import math

def main():
    while True:
        user_input = input("输入一个非负整数（输入'off'终止程序）：").strip()
        if user_input.lower() == 'off':
            print("计算已终止")
            break
        try:
            base = int(user_input)
            if base < 0:
                print("阶乘仅支持非负整数，请重新输入")
                continue
            result = math.factorial(base)
            print(f"{base}的阶乘是：{result}")
        except ValueError:
            print("输入无效，请输入非负整数或'off'")

if __name__ == "__main__":
    main()

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