面试题 08.01. 三步问题

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

示例1:

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

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

n范围在[1, 1000000]之间

解题思路（动态规划）：

要想走上当前第n阶台阶，可以选择从第n-1、第n-2和第n-3阶台阶上楼梯，所以上楼梯的方式是上述累加和

代码实现：

class Solution {
public:
    int waysToStep(int n) {
        if(n<=2){
            return n;
        }
        if(n==3){
            return 4;
        }
        int* dp=new int[n+1];
        dp[0]=0;
        dp[1]=1;
        dp[2]=2;
        dp[3]=4;
        int mod=1000000007;
        for(int i=4;i<=n;++i){
            dp[i]=((dp[i-1]%mod+dp[i-2]%mod)%mod+dp[i-3]%mod)%mod;
        }
        return dp[n];
    }
};