leetcode-dp-三步问题

package dp.waysToStep;

/**
 * 面试题 08.01. 三步问题
 * 三步问题。有个小孩正在上楼梯，楼梯有n阶台阶，小孩一次可以上1阶、2阶或3阶。实现一种方法，计算小孩有多少种上楼梯的方式。结果可能很大，你需要对结果模1000000007。
 * <p>
 * 示例1:
 * <p>
 * 输入：n = 3
 * 输出：4
 * 说明: 有四种走法
 * 示例2:
 * <p>
 * 输入：n = 5
 * 输出：13
 * 提示:
 * <p>
 * n范围在[1, 1000000]之间
 */
public class waysToStep {
    public static int waysToStep(int n) {
        int size = n<3?3:n;
        int[] dp = new int[size];
        dp[0] = 1;
        dp[1] = 2;
        dp[2] = 4;
        for (int i = 3; i < n; i++) {
            //dp[i-3]上一步已经取模了，剩下俩个防止溢出也要取模
            dp[i] = (dp[i - 1] + dp[i - 2]) % 1000000007 + dp[i - 3];
            dp[i] %= 1000000007;
        }
        return dp[n-1];
    }

    public static void main(String[] args) {
        System.out.println(waysToStep(1));
    }
}