三步问题。有个小孩正在上楼梯,楼梯有n阶台阶,小孩一次可以上1阶、2阶或3阶。实现一种方法,计算小孩有多少种上楼梯的方式。
解题思路
dp[i]含义:走到i阶台阶一共有dp[i]种方式
dp数组初始化:
dp[0]:0层台阶?没有意义,不用多管,直接赋值为null
dp[1]:1层台阶,只有1种方式
dp[2]:2层台阶,有1+1、2一共2种方式,故dp[2]=2
dp[3]:3层台阶,有1+1+1、1+2、2+1、3一共4种方式,故dp[3]=4
所以dp = [null, 1, 2, 4]
递推公式:第i层台阶可以由i-1、i-2、i-3层台阶相加而来,取模,故dp[i] = (dp[i - 1] + dp[i - 2] + dp[i - 3])
遍历:前3层都已经确定了,所以从i=4开始遍历,从前向后遍历。
代码
var waysToStep = function(n) {
dp = [null,1,2,4];
for(let y= 4;y<=n;y++){
dp[y]=(dp[y-3]+dp[y-2]+dp[y-1])%
}
return dp[n]
};