leetcode 面试题 08.01. 三步问题

最新推荐文章于 2024-06-22 19:46:49 发布

Hag_20

最新推荐文章于 2024-06-22 19:46:49 发布

阅读量339

点赞数

文章标签： leetcode 算法动态规划

本文链接：https://blog.csdn.net/qq_62623517/article/details/122893238

版权

题目描述：

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

思路：

很简单的dp，关键在于取模结合加法运算的处理。

由公式(a+b)%m=(a%m+b%m)%m可推得(a+b+c)%m=[(a%m+b%m)%m+c%m]%m.

代码：

int waysToStep(int n){
    int dp[1000001]={1,1,2,4};
    for(int i=4;i<=n;i++){
        int t1=dp[i-3]%1000000007;int t2=dp[i-2]%1000000007;
        int sum1=(t1+t2)%1000000007;
        int t3=dp[i-1]%1000000007;
        dp[i]=(sum1+t3)%1000000007;
    }
    return dp[n];
}