问题描述:三步问题。有个小孩正在上楼梯,楼梯有n阶台阶,小孩一次可以上1阶、2阶或3阶。实现一种方法,计算小孩有多少种上楼梯的方式。结果可能很大,你需要对结果模1000000007。
示例1:
输入:n = 3
输出:4
说明: 有四种走法
示例2:
输入:n = 5
输出:13
提示:
n范围在[1, 1000000]之间
动态规划求解代码:
#define INF 1000000007
class Solution {
public:
int waysToStep(int n) {
int s[1000005];
s[1]=1;
s[2]=2;
s[3]=4;
for(int i=4;i<=n;i++)
{
s[i]=((s[i-1]+s[i-2])%INF+s[i-3]%INF)%INF;//多次求余避免溢出
}
return s[n];
}
};
结果: