程序员面试金典08.01-三步问题

原题(来源Leetcode)

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

示例1:

输入: n = 3
输出: 4
说明: 有四种走法

示例2:

输入: n = 5
输出: 13

提示:

  1. n范围在[1, 1000000]之间

解法

思想

动态规划,注意溢出的问题。

代码

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

原创文章,作者:彭晨涛,如若转载,请注明出处:https://www.codetool.top/article/%e7%a8%8b%e5%ba%8f%e5%91%98%e9%9d%a2%e8%af%95%e9%87%91%e5%85%b808-01-%e4%b8%89%e6%ad%a5%e9%97%ae%e9%a2%98/

发表回复

登录后才能评论