剑指 Offer 10- I. 斐波那契数列

写一个函数，输入 n ，求斐波那契（Fibonacci）数列的第 n 项（即 F(N)）。斐波那契数列的定义如下：

F(0) = 0，F(1) = 1
F(N) = F(N - 1) + F(N - 2)，其中 N > 1.

斐波那契数列由 0 和 1 开始，之后的斐波那契数就是由之前的两数相加而得出。
答案需要取模 1e9+7（1000000007），如计算初始结果为：1000000008，请返回 1。

示例 1：

输入：n = 2
输出：1

示例 2：

输入：n = 5
输出：5

方法：动态规划

• 状态定义：$dp[i]$为斐波那契数列的第i项
• 转移方程：$dp[i+2]=dp[i]+dp[i+1]$
• 初始状态： $dp[0]=0，dp[1]=1$
• 返回值：$dp[]$数组最后一个数

Python3

class Solution:
    def fib(self, n: int) -> int:
        if n == 0: return 0
        dp =[0]*(n+1)
        dp[1] = 1
        for i in range(n-1):
            dp[i+2] = dp[i] + dp[i+1]
        return dp[-1] % 1000000007
        

代码优化

此时空间复杂度为 $O(N)$。

• 由于$dp[]$列表第i项只与第 $i-1$ 和第 $i-2$ 项有关，因此只需要初始化三个整形变量$sum,a,b$，利用辅助变量 $sum$使$a,b$两数字交替前进即可。
• 节省了$dp[]$列表空间，因此空间复杂度降至$O(1)$。

改进代码

class Solution:
    def fib(self, n: int) -> int:
        a, b = 0, 1
        for _ in range(n):
            a, b = b, a + b
        return a % 1000000007