面試題10 - II.青蛙跳臺階問題

面試題10 - II.青蛙跳臺階問題

題目描述

一隻青蛙一次可以跳上1級臺階，也可以跳上2級臺階。求該青蛙跳上一個 n 級的臺階總共有多少種跳法。

答案需要取模 1e9+7（1000000007），如計算初始結果爲：1000000008，請返回 1。

示例 1：

輸入：n = 2
輸出：2
示例 2：

輸入：n = 7
輸出：21
提示：

0 <= n <= 100

題解

n個臺階，若第一次跳一階則剩下n-1臺階，若第一次跳兩階則剩下n-2階；容易得到：f（n）=f（n-1）+f（n-2），代碼如下：

class Solution {
    public int numWays(int n) {
        int a = 1, b = 1, sum;
        for(int i = 0; i < n; i++){
        //這裏要對sum值取模保證結果在int範圍內，因爲int範圍是-2147483648～2147483647；
        //是十位數，那麼取模十位最小質數1000000007保證結果在int範圍內
            sum = (a + b)% 1000000007;
            a = b;
            b = sum;
        }
        return a;
    }
}

提交結果