题目
一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。
答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。
示例 1:
输入:n = 2
输出:2
示例 2:
输入:n = 7
输出:21
示例 3:
输入:n = 0
输出:1
提示:
- 0 <= n <= 100
解题
简单的分析,可知跳台阶问题是斐波那切数列问题,跳n个台阶的方式就是斐波那切数列n项的值。那么问题就很简单了,可以用动态规划求解该问题。
解法1
动态一维数组,记录前两个的值。
时间复杂度:O(n)
空间复杂度:O(n)
public int numWays(int n) {
if (n == 0 || n == 1) return 1;
int[] res = new int[n+1];
res[0] = 1;
res[1] = 1;
for (int i = 2; i < res.length; i++) {
res[i] = (res[i-1] + res[i-2])%1000000007;
}
return res[n];
}
解法2
时间复杂度:O(n)
空间复杂度:O(1) 没有用一维数组,而是使用变量的内存空间;
public int numWays(int n) {
if (n < 2) return 1;
int first = 1;
int second = 1;
int sum = 1;
for (int i = 2; i <= n; i++) {
sum = (first+second)%1000000007;
first = second;
second = sum;
}
return sum;
}
注意:为什么需要用1000000007取模?大数的排列组合等为什么要取模?
- 1000000007是一个质数(素数),对质数取余能最大程度避免结果冲突/重复
- int32位的最大值为2147483647,所以对于int32位来说1000000007足够大。
- int64位的最大值为2^63-1,用最大值模1000000007的结果求平方,不会在int64中溢出。
所以在大数相乘问题中,因为(a∗b)%c=((a%c)∗(b%c))%c,所以相乘时两边都对1000000007取模,再保存在int64里面不会溢出。