[Leetcode]面试题08.01三步问题

题目描述

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

示例1:

输入：n = 3
输出：4
说明: 有四种走法
示例2:

输入：n = 5
输出：13
提示:

n范围在[1, 1000000]之间

题目分类

动态规划

思路

递推式res[i] = res[i-1] + res[i-2] + res[3];

代码

class Solution {
public:
    int waysToStep(int n) {
        if(n==0) return 0;
        else if(n==1) return 1;
        else if(n==2) return 2;
        else if(n==3) return 4;
        else {
            long a = 1, b = 2, c = 4, res = 4;
            for(int i = 3; i < n; i++){
                res = (a+b+c) % 1000000007;
                a = b % 1000000007;
                b = c % 1000000007;
                c = res % 1000000007;
            }
            return res;
        }
    }
};