[动态规划]LeetCode 面试题 08.01. 三步问题
三步问题。有个小孩正在上楼梯,楼梯有n阶台阶,小孩一次可以上1阶、2阶或3阶。实现一种方法,计算小孩有多少种上楼梯的方式。结果可能很大,你需要对结果模1000000007。
示例1:
输入:n = 3
输出:4
说明: 有四种走法
示例2:
输入:n = 5
输出:13
提示:
- n范围在[1, 1000000]之间
最基础滴
f(i)表示i阶台阶的走法
class Solution {
public:
int waysToStep(int n) {
vector<int> f(n + 1);
if (n <= 2) return n;
f[0] = 0,f[1] = 1,f[2] = 2,f[3] = 4;
int mod = 1000000007;
for(int i = 4; i <= n; ++i)
f[i] = (long long)((long long)(f[i - 1] + f[i - 2])% mod + f[i - 3]) % mod;
return f[n];
}
};