面试题10- II

一只青蛙一次可以跳上1级台阶,也可以跳上2级台阶。求该青蛙跳上一个 n 级的台阶总共有多少种跳法。

答案需要取模 1e9+7(1000000007),如计算初始结果为:1000000008,请返回 1。

示例 1:

输入:n = 2
输出:2

示例 2:

输入:n = 7
输出:21

提示:

  • 0 <= n <= 100

注意:本题与主站 509 题相同:https://leetcode-cn.com/problems/fibonacci-number/

Solutions

  1. dynamic programming

  2. dp[i] = dp[i - 1] + dp[i - 2]

  3. Base case: dp[0] = 1, dp[1] = 1

class Solution {
public:
    int fib(int n) {
        if (!n) return 0;
        n--;
        int pprev = 0, prev = 1;
        while (n--) {
            prev += pprev;
            pprev = prev - pprev;
            prev %= 1000000007;
        }

        return prev;
    }

    int numWays(int n) {
        return fib(n + 1);    
    }
};

Last updated

Was this helpful?