剑指 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为一维数组，其中 dp[i]的值代表 斐波那契数列第 i个数字 。
转移方程： dp[i + 1] = dp[i] + dp[i - 1] ，即对应数列定义 f(n + 1) = f(n) + f(n - 1) ；
初始状态： dp[0] = 0, dp[1] = 1，即初始化前两个数字；
返回值： dp[n] ，即斐波那契数列的第 n 个数字。

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

代码如下

class Solution {
public:
    int fib(int n) {
        if(n==0)
        {
            return 0;
        }
        if(n==1)
        {
            return 1;
        }
        int a=0;
        int b=1;
        int sum=0;
        for(int i=2; i<=n; ++i){
            sum=(a+b)%1000000007;
            a=b;
            b=sum;
        }
        return sum;
    }
};