写一个函数,输入 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
提示:0 <= n <= 100
思路
典型的DP。设dp[i]表示N为i时的答案。通过递推公式dp[i]=dp[i-1]+dp[i-2]进行计算即可。
使用两个变量交替表达dp[i-1]和dp[i-2]可以省区vector的空间占用
答案
class Solution {
public:
int fib(int n) {
vector<int> dp(n+2,0);
dp[1]=1;
for(int i=2;i<=n;i++) {
dp[i]=(dp[i-1]+dp[i-2])%1000000007;
}
return dp[n];
}
};
class Solution {
public:
int fib(int n) {
if(n==0) return 0;
if(n==1) return 1;
int a=0,b=1,c=0; //c=b+a
for(int i=2;i<=n;i++) {
c=(a+b)%1000000007;
a=b;
b=c;
}
return c;
}
};