[动态规划]LeetCode 面试题 08.01. 三步问题

[动态规划]LeetCode 面试题 08.01. 三步问题

三步问题。有个小孩正在上楼梯,楼梯有n阶台阶,小孩一次可以上1阶、2阶或3阶。实现一种方法,计算小孩有多少种上楼梯的方式。结果可能很大,你需要对结果模1000000007。

示例1:

 输入:n = 3 
 输出:4
 说明: 有四种走法

示例2:

 输入:n = 5
 输出:13

提示:

  1. 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];
    }
};

你可能感兴趣的:(LeetCode)