每日力扣-动态规划-三步问题

94 阅读1分钟

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

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

输入:n = 5 输出:13

解题思路:考虑第i项走法,可以是第i-3跳三格,或者i-2跳两格,或者i-1跳一格,所以第i项走法是这3个相加。

 class Solution {
public:
    int waysToStep(int n) {
        if(n <= 2) return n;
        if(n == 3) return 4;
        vector<long> dp(n+2 , 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 dp[n];
    }
};