斐波那契数(简单)
斐波那契数 (通常用 F(n)
表示)形成的序列称为 斐波那契数列 。该数列由 0
和 1
开始,后面的每一项数字都是前面两项数字的和。也就是:
F(0) = 0,F(1) = 1 F(n) = F(n - 1) + F(n - 2),其中 n > 1
给定 n
,请计算 F(n)
。
解法一:动态规划
思路与70、爬楼梯一致,利用的公式F(n) = F(n - 1) + F(n - 2)
class Solution {
public:
int fib(int n) {
if(n<2){
return n;
}
vector<int> f(n+1);
f[0]=0;
f[1]=1;
for (int i=2;i<=n;i++){
f[i]=f[i-1]+f[i-2];
}
return f[n];
}
};
或者换一种思路,将斐波那契数列看做窗口大小为3的滑动窗口 ,可以将空间复杂度降低为O(1)
class Solution {
public:
int fib(int n) {
if (n < 2) {
return n;
}
int p = 0, q = 0, r = 1;
for (int i = 2; i <= n; ++i) {
p = q;
q = r;
r = p + q;
}
return r;
}
};
解法二:矩阵快速幂
class Solution {
public:
int fib(int n) {
if (n < 2) {
return n;
}
//定义基底(1,1)(1,0)
vector<vector<int>> q{{1, 1}, {1, 0}};
//幂次为n-1
vector<vector<int>> res = matrix_pow(q, n - 1);
return res[0][0];
}
vector<vector<int>> matrix_pow(vector<vector<int>>& a, int n) {
vector<vector<int>> ret{{1, 0}, {0, 1}};
//1.若为奇数,需多乘一次基底
//2.若幂次除到1,乘积后得到res
//此处使用位运算在于效率高
while (n > 0) {
if (n & 1) {
ret = matrix_multiply(ret, a);
}
n >>= 1;
a = matrix_multiply(a, a);
}
return ret;
}
vector<vector<int>> matrix_multiply(vector<vector<int>>& a, vector<vector<int>>& b) {
vector<vector<int>> c{{0, 0}, {0, 0}};
for (int i = 0; i < 2; i++) {
for (int j = 0; j < 2; j++) {
c[i][j] = a[i][0] * b[0][j] + a[i][1] * b[1][j];
}
}
return c;
}
};