题目一:求斐波那契数列的第n项
写一个函数,输入 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)定义dp数组,
dp[i]
即F(N)
- 2)状态转移方程:
dp[i] = dp[i - 1] + dp[i - 2]
- 3)初始化数组:
dp[0] = 0
,dp[1] = 1
class Solution {
public int fib(int n) {
if (n == 0) return 0;
if (n == 1) return 1;
int p1 = 0, p2 = 1, p3 = 0;
for (int i = 2; i <= n; i++) {
p3 = (p1 + p2) % 1000000007;
p1 = p2;
p2 = p3;
}
return p3;
}
}
复杂度分析
- 时间复杂度
O(N)
- 空间辅助度
O(1)
题目二: 青蛙跳台问题
一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
输入:n = 0
输出:1
思路
当n为1的时候,有1种走法,n为2的时候有两种走法,n为3的时候有3种,n为4的时候有5种
1: 1
2: 2
3: f(1) + f(2)
4: f(3) + f(2)
f(n): f(n-1) + f(n-2)
分析完之后就是一个斐波拉契数列
-
那么就可以用迭代的方法来解:
class Solution { public int numWays(int n) { if (n == 0) return 1; if (n == 1) return 1; if (n == 2) return 2; int p1 = 1, p2 = 2, p3 = 0; for (int i = 3; i <= n; i++) { p3 = (p1 + p2) % 1000000007; p1 = p2; p2 = p3; } return p3; } }
class Solution {
public int numWays(int n) {
int p = 0, q = 0, r = 1;
for (int i = 1; i <= n; ++i) {
p = q;
q = r;
r = (p + q) % 1000000007;
}
return r;
}
}
复杂度分析
- 时间复杂度
O(n)
- 空间复杂度
O(1)