Python斐波那契数列实现：优化递归与边界处理 – DeepSeek 代码改写提示词实例

对代码进行修改，来实现纠错、注释、调优等。

提示词

下面这段的代码的效率很低，且没有处理边界情况。请先解释这段代码的问题与解决方法，然后进行优化：
```
def fib(n):
    if n <= 2:
        return n
    return fib(n-1) + fib(n-2)
```

样例输出

### 问题分析

这段代码实现了一个递归的斐波那契数列计算函数 `fib(n)`。虽然代码简洁，但它存在以下几个问题：

1. **重复计算**：递归调用会导致大量的重复计算。例如，计算 `fib(5)` 时，`fib(3)` 和 `fib(2)` 会被多次计算，导致效率低下。

2. **边界情况处理不完善**：虽然代码中处理了 `n <= 2` 的情况，但没有处理 `n < 0` 的情况。对于负数输入，函数会返回不正确的结果。

3. **递归深度限制**：对于较大的 `n`，递归深度可能会超过 Python 的默认递归深度限制，导致 `RecursionError`。

### 解决方法

1. **使用记忆化（Memoization）**：通过缓存已经计算过的斐波那契数列值，避免重复计算，提高效率。

2. **处理边界情况**：增加对 `n < 0` 的检查，并返回适当的错误信息或默认值。

3. **使用迭代代替递归**：递归虽然简洁，但在处理大规模数据时效率较低。可以使用迭代方法来计算斐波那契数列，避免递归深度限制。

### 优化后的代码

#### 方法一：记忆化递归

```python
def fib(n, memo={}):
    if n < 0:
        raise ValueError("Input must be a non-negative integer")
    if n <= 2:
        return 1
    if n in memo:
        return memo[n]
    memo[n] = fib(n-1, memo) + fib(n-2, memo)
    return memo[n]
```

#### 方法二：迭代法

```python
def fib(n):
    if n < 0:
        raise ValueError("Input must be a non-negative integer")
    if n <= 2:
        return 1
    
    a, b = 1, 1
    for _ in range(3, n+1):
        a, b = b, a + b
    return b
```

### 解释

1. **记忆化递归**：
   - 使用一个字典 `memo` 来存储已经计算过的斐波那契数列值。
   - 在每次递归调用前，先检查 `memo` 中是否已经有结果，如果有则直接返回，避免重复计算。
   - 处理了 `n < 0` 的情况，抛出 `ValueError`。

2. **迭代法**：
   - 使用两个变量 `a` 和 `b` 来保存前两个斐波那契数列值。
   - 通过循环从 `3` 到 `n` 计算斐波那契数列值，避免了递归调用，提高了效率。
   - 同样处理了 `n < 0` 的情况，抛出 `ValueError`。

### 总结

优化后的代码不仅提高了效率，还完善了边界情况的处理。记忆化递归和迭代法都是有效的优化手段，具体选择哪种方法取决于具体需求和场景。

代码

from openai import OpenAI

client = OpenAI(
    base_url="https://api.deepseek.com/",
    api_key="<YOUR_API_KEY>"
)

completion = client.chat.completions.create(
    model="deepseek-chat",
    messages=[
        {
                "role": "user",
                "content": "下面这段的代码的效率很低，且没有处理边界情况。请先解释这段代码的问题与解决方法，然后进行优化：\n```\ndef fib(n):\n    if n <= 2:\n return n\n return fib(n-1) + fib(n-2)\n```" } ] ) print(completion.choices[0].message.content)

◀返回【DeepSeek 官方提示词大全】查看更多。