[算法练习及思路-程序员面试金典(Java解法)]No41.三步问题

题号:no41

题目名:三步问题

原题URL:https://leetcode-cn.com/problems/three-steps-problem-lcci/

题目描述

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

示例

示例 1：

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

示例 2：

 输入：n = 5
 输出：13

限制

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

思路

1.简单的动态规划，和青蛙跳楼一样

2.注意重点就是mod的运算，只有两数相加的时候才有可能超过100000007，这时候Mod可以让数变小

解题代码

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