题目描述
写一个函数,输入 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。
解法一:动态规划
由于斐波那契数存在递推关系,因此可以使用动态规划求解。动态规划的状态转移方程即为上述递推关系,边界条件为 F(0)和 F(1)。
根据状态转移方程和边界条件,可以得到时间复杂度和空间复杂度都是 O(n)的实现。由于 F(n)只和 F(n−1) 与 F(n−2) 有关,因此可以使用「滚动数组思想」把空间复杂度优化成 O(1)。如下的代码中给出的就是这种实现。
class Solution {
public int fib(int n) {
int dp1=0, dp2=1, dp=0;
if(n==0 || n==1) return n;
else {
for (int i=2; i<=n; i++) {
dp = (dp1 + dp2) % 1000000007;
dp1 = dp2;
dp2 = dp;
}
return dp;
}
}
}
解法2:矩阵快速幂
方法一的时间复杂度是O(n)。使用矩阵快速幂的方法可以降低时间复杂度。
因此只要我们能快速计算矩阵 M的 n次幂,就可以得到 F(n) 的值。如果直接求取
M
n
M^n
Mn,时间复杂度是O(n),可以定义矩阵乘法,然后用快速幂算法来加速这里
M
n
M^n
Mn的求取。
所谓矩阵快速幂,其原理参见数论-快速幂、矩阵快速幂
class Solution {
static final int MOD = 1000000007;
public int fib(int n) {
if (n < 2) {
return n;
}
int[][] q = {{1, 1}, {1, 0}};
int[][] res = pow(q, n - 1);
return res[0][0];
}
public int[][] pow(int[][] a, int n) {
int[][] ret = {{1, 0}, {0, 1}};
while (n > 0) {
if ((n & 1) == 1) {
ret = multiply(ret, a);
}
n >>= 1;
a = multiply(a, a);
}
return ret;
}
public int[][] multiply(int[][] a, int[][] b) {
int[][] c = new int[2][2];
for (int i = 0; i < 2; i++) {
for (int j = 0; j < 2; j++) {
c[i][j] = (int) (((long) a[i][0] * b[0][j] + (long) a[i][1] * b[1][j]) % MOD);
}
}
return c;
}
}