Triple Step:一个小孩要爬n
级台阶,每次可以上1
、2
和3
步,计算上n
级台阶一共有多少种方法。
最后一级台阶可以通过3
种方法到达:
- 在第
n - 1
级时上1
步 - 在第
n - 2
级时上2
步 - 在第
n - 3
级时上3
步
最后把这3
种情况的方法求和即可,即f(n) = f(n - 1) + f(n - 2) + f(n - 3)
,类似斐波那契数列,时间复杂度为O(3 ^ n)
,因为每一项大概需要计算3
次。同时还要注意边界条件,这里定义f(0) = 1
,f(-1) = 0
,f(-2) = 0
,因此f(1)
也可以通过该公式计算。
为了减少计算次数,可以将计算结果存下来,等到需要的时候直接读取即可,也就是记忆化搜索,这种方法应该不如递推快。
class Solution {
public:
int waysToStep(int n) {
Ways.assign(n + 1, 0);
Ways[0] = 1;
return countWay(n);
}
private:
const int MOD = 1000000007;
vector<int> Ways;
int countWay(int n)
{
if(n < 0) return 0;
else if(Ways[n] != 0) return Ways[n];
else{
Ways[n] = (countWay(n - 1) + countWay(n - 2)) % MOD;
Ways[n] += countWay(n - 3);
Ways[n] %= MOD;
return Ways[n];
}
}
};