剑指 10 斐波那契数列
原题目
写一个函数,输入 n
,求斐波那契(Fibonacci)数列的第 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
自己的第一遍解法
看到表达式的第一眼当然想到递归啊,剑指书上也是直接就给出了递归的代码。
int fib
{
if (n<=0) return 0;
if (n==1) return 1;
return fib(n-1) + fib(n-2);
}
虽然用递归的代码比较简洁,但是从上到下的递归里面包含了大量的重复计算,时间复杂度以n的指数次幂递增。
好的解法
这一题好的解法是使用递归的思想,但是使用循环来解题。
既然f(n)
只与f(n-1)
、f(n-2)
有关系,那么就只保存这三个数,每次更新这三个数就会得到最新的f(n)
class Solution {
public:
int fib(int n) {
int res[2] = {0, 1};
if (n<2)
return res[n];
int fn_1 = 1;
int fn_2 = 0;
int f_n = 0;
for (int i=2; i<=n; ++i)
{
f_n = (fn_1 + fn_2)%1000000007; //题目要求取模
fn_2 = fn_1; //注意一定先更新时间最久远的变量
fn_1 = f_n;
}
return f_n;
}
};
使用循环,保存前两次计算的值,这样少去了很多重复计算,大大提高了时间效率
斐波那契数列变形:青蛙跳台阶
原题目
一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
示例 1:
输入:n = 2
输出:2
示例 2:
输入:n = 7
输出:21
示例 3:
输入:n = 0
输出:1
解法
遇到这种问题很大不好想的时候,要从最简单的情况开始:n=1 时,只有 1 种跳法;n=2 时,有两种跳法;n>2 时,第一次跳有两种方法,一种是跳 1 阶,此时跳法数量 = 后面 n-1阶台阶的跳法,还有一种就是第一次跳 2 阶,此时跳法数量 = 后面n-2阶台阶的跳法。因此在 n>2 时,跳法数量就是一个斐波那契数列。
class Solution {
public:
int numWays(int n) {
int res[3] = {1, 1, 2};
if (n<=2)
return res[n];
int fn_1 = 2;
int fn_2 = 1;
int fn = 0;
for (int i=3; i<=n; ++i)
{
fn = (fn_1 + fn_2)%1000000007;
fn_2 = fn_1;
fn_1 = fn;
}
return fn;
}
};