面试题 08.01. 三步问题
三步问题。有个小孩正在上楼梯,楼梯有n阶台阶,小孩一次可以上1阶、2阶或3阶。实现一种方法,计算小孩有多少种上楼梯的方式。结果可能很大,你需要对结果模1000000007。
示例1:
输入:n = 3
输出:4
说明: 有四种走法
示例2:
输入:n = 5
输出:13
提示:
n范围在[1, 1000000]之间
class Solution {
public int waysToStep(int n) {
if(n == 1)
return 1;
if(n == 2)
return 2;
if(n == 3)
return 4;
long dp[] = new long[n+1];
dp[0] = 0;
dp[1] = 1;
dp[2] = 2;
dp[3] = 4;
for(int i = 4 ; i <= n ; i++){
dp[i] = (dp[i-1] + dp[i-2] + dp[i-3])%1000000007;
}
return (int)dp[n];
}
}
笔记
- 动态规划的基础题,画出递归树的话,想想怎么构造dp数组,递归方程在纸上随便写写就能发现规律。
- 大整数取模问题挺有意思的,不过问题太特殊,一般也可以用长整型解决。
总结
超大数记得取模1000000007,有空看看原理。