今天来给大家介绍几个在面试中常用的、求解斐波那契数列问题的方法。
首先定义斐波那契数列问题:
定义 a0=1, a1=1, an=an−1+an−2,求 an 是多少。
为了避免考虑整数溢出问题,我们求 an%p 的值,p=109+7。
算法1
递归。
递归计算的节点个数是 O(2n) 的级别的,存在大量重复计算。
时间复杂度是 O(2n),一秒内大约能算到第三四十项。
C++ 代码
const int MOD = 1000000007;
int f(int n)
{
if (n <= 1) return 1;
return (f(n - 1) + f(n - 2)) % MOD;
}
算法2
记忆化搜索。
开一个大数组记录中间结果,如果一个状态被计算过,则直接查表,否则再递归计算。
总共有 n 个状态,计算每个状态的复杂度是 O(1),所以时间复杂度是 O(n)。
一秒内算 n=107 毫无压力,但由于是递归计算,递归层数太多会爆栈,大约只能算到 n=105 级别。
C++ 代码
const int N = 100000, MOD = 1000000007;
int a[N];
int f2(int n)
{
if (a[n]) return a[n];
if (n <= 1) return 1;
a[n] = f2(n - 1) + f2(n - 2);
a[n] %= MOD;
return a[n];
}
算法3
递推。
开一个大数组,记录每个数的值。用循环递推计算。
总共计算 n 个状态,所以时间复杂度是 O(n)。
但需要开一个长度是 n 的数组,内存将成为瓶颈,当 n=108 时,需要的内存是 4∗1081024×1024≈381MB。
分子中乘4是因为C++中 int
类型占4字节。
C++代码
const int N = 100000000, MOD = 1000000007;
int f3(int n)
{
a[0] = a[1] = 1;
for (int i = 2; i <= n; i ++ )
{
a[i] = a[i - 1] + a[i - 2];
a[i] %= MOD;
}
return a[n];
}
算法4
递归+滚动变量。
仔细观察我们会发现,递推时我们只需要记录前两项的值即可,没有必要记录所有值,所以我们可以用滚动变量递推。
时间复杂度还是 O(n),但空间复杂度变成了 O(1)。
C++代码:
const int MOD = 1000000007;
int f4(int n)
{
if (n <= 1) return 1;
int x, y, z;
x = y = 1;
for (int i = 2; i <= n; i ++ )
{
z = (x + y) % MOD;
x = y;
y = z;
}
return z;
}
算法5
矩阵运算 + 快速幂。
快速幂算法的模板可以参考这里。
用算法4我们1秒内最多可以算到 108 级别,那当 n 更大时该怎么办呢?
可以先利用矩阵运算的性质将通项公式变成幂次形式,然后用平方倍增(快速幂)的方法求解第 n 项。
首先我们定义向量
Xn=[anan−1]
边界:
X1=[a1a0]
然后我们可以找出矩阵:
A=[1110]
则有:
Xn=Xn−1×A
所以:
Xn=X1×An−1
由于矩阵具有结合律,所以我们可以先求出 An−1%P,然后再用 X1 左乘,即可求出 Xn,向量 Xn 的第一个元素就是 an。
时间复杂度分析:快速幂的时间复杂度是 O(logn),所以算法5的时间复杂度也是 O(logn)。
C++代码
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <ctime>
using namespace std;
const int MOD = 1000000007;
void mul(int a[][2], int b[][2], int c[][2])
{
int temp[][2] = {{0, 0}, {0, 0}};
for (int i = 0; i < 2; i ++ )
for (int j = 0; j < 2; j ++ )
for (int k = 0; k < 2; k ++ )
{
long long x = temp[i][j] + (long long)a[i][k] * b[k][j];
temp[i][j] = x % MOD;
}
for (int i = 0; i < 2; i ++ )
for (int j = 0; j < 2; j ++ )
c[i][j] = temp[i][j];
}
int f_final(long long n)
{
if (!n) return 1;
int x[2] = {1, 1};
int res[][2] = {{1, 0}, {0, 1}};
int t[][2] = {{1, 1}, {1, 0}};
long long k = n - 1;
while (k)
{
if (k&1) mul(res, t, res);
mul(t, t, t);
k >>= 1;
}
int c[2] = {0, 0};
for (int i = 0; i < 2; i ++ )
for (int j = 0; j < 2; j ++ )
{
long long r = c[i] + (long long)x[j] * res[j][i];
c[i] = r % MOD;
}
return c[0];
}
int main()
{
long long n ;
cin >> n;
cout << f_final(n) << endl;
return 0;
}
算法4有一丁点的小问题,就是当n == 0 || n == 1的时候 z 木有值
然后算法5 牛逼!
多谢指正,代码已修正。
#include <bits/stdc++.h> using namespace std; int main() { double t = sqrt(5); int n1;cin >> n1; while (n1 -- ){ int n;cin>>n; double fib_n = 1.0 / t * ((pow((1 + t) / 2.0, n) - pow((1 - t) / 2.0, n))); cout << "Fib(" << n<<") = " << (long long)fib_n << endl; } return 0; } 这个公式也可以很快算出来嘿嘿
算法4是递推+滚动变量吧
对哦
记忆化搜索是不是有点动态规划的思想在里头
记忆化搜索是dp的一个分支
为什么暴力递归的时间复杂度是O(n)啊?
算法3的代码中缺少了 int a[N];
int x[2] = {1, 1};
int res[][2] = {{1, 0}, {0, 1}};
这个初值是什么意思?
算法5 那个 int x[2] = {1, 1};是不是有点问题呀 不应该是 int x[2] = {1, 0};嘛 按照上面的算 f_final(2)就等于2了
感觉老师这里的初始值是这样的f[1] = f[0] = 1
记忆化搜索,数组好像也开不了太大
最后一步 xn=x1∗An−1,代码里 x[2] 为啥不等于 {1, 0}
看定义是 a[1] = a[0] = 1
再分享一种方法,更简单高效的方法,就是直接把菲波那切数列的通项公式解出来。
int Fibonacci(int n) {
double sqrt5 = sqrt(5);
double alpha = (1 + sqrt5) / 2;
double beta = (1 - sqrt5) / 2;
double ret = (pow(alpha, n) - pow(beta, n)) / sqrt5;
return (int) ret;
}
但是精度不高啊
还好,是可以ac的,只是n不能太大
万一公式记错了怎么办
学一下特征方程 再现场推
简单是简单,但是没有高效吧
1. 从时间复杂度来说:pow即使是快速幂也是log(n)复杂度,没有变
2. double的精度没有很高
3. 如何%p?代码中好像没有看到任何关于取模p的地方
tql
hh,怎么感觉算法4应该是递推+滚动变量
AcWing 1213 怎么做?
算法5
f_final
里面的a[2][2]
好像是多余的是的hh,已经去掉了
@y总,算法4应该是 x = 0, y =1;
x和y分别表示第0项和第1项,这里统一定义为1。
时间复杂度O(logn)
是指 以 10 为底 n 的对数吗?
如果深究应该是以2为底,但是时间复杂度这个概念不在乎系数,比如n和2n都是O(n)。由于对数运算的性质log_{a}^{b}✖log_{b}^{n}=log_{a}^{n}所以一般不再说以什么为底,就是O(logn)
log_{a}^{b}是个常数
哦,好的,明白
算法时间复杂度为log(n)时,不同底数对应的时间复杂度的倍数关系为常数,不会随着底数的不同而不同,因此可以将不同底数的对数函数所代表的时间复杂度,当作是同一类复杂度处理
对滴。算法复杂度中一般以2为底。
y中 这一步Xn=X1×An−1 对应的代码 long long r = c[i] + (long long)x[j] * res[i][j]; 这个res[i][j] 的i和j是不是反了 变成行乘行了 应该行乘列才对吧
多谢指正~ 已更正。之前X恰好是[1, 1] 且 res 恰好是对称矩阵,所以答案恰好是对的,但过程是错的。
y总,返回的c[0]好像是第 n + 1项,c[1]才是第 n 项
1 1 2 3 5 8
n = 5 时得到的 是 8
最前面的定义是:
所以 a_5 的值在这种定义下就是8。
太强了
谢谢hh