[Java 解题系列] LeetCode: 面试题 08.01. 三步问题

题目

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

示例 1:

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

示例 2:

输入：n = 5
输出：13

提示:

n 范围在 [1, 1000000] 之间

分析

• 1 个台阶的情况（1 种方式）

  • 上 1 阶

• 2 个台阶的情况（2 种方式）

  • 先上到台阶 1，再上 1 阶
  • 直接上 2 阶

• 3 个台阶的情况

  • 先上到台阶 1，再上 2 阶（1 种方式）
  • 先上到台阶 2，再上 1 阶（2 种方式）
  • 直接上 3 阶

• 所有 n>3 阶的楼梯，有如下三类方式：

  • 先上到台阶 n-3，再上 3 阶
  • 先上到台阶 n-2，再上 2 阶
  • 先上到台阶 n-1，再上 1 阶

我们使用 dp[i] 表示上到第 i 阶台阶的方式
则有 dp[i] = dp[i-3]+dp[i-2]+dp[i-3] (i>3);

题解

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