剑指Offer10-2.青蛙跳台阶问题Golang版

剑指Offer10-2.青蛙跳台阶问题Golang版

1. 问题描述

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

答案需要取模 1e9+7（1000000007），如计算初始结果为：1000000008，请返回 1。

示例 1：

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

输入：n = 7
输出：21
示例 3：

输入：n = 0
输出：1

2. 思路

动态规划思想，

1. 1层台阶，1种
2. 2层台阶，2种
3. 3层台阶，3种
4. 4层台阶，5种
5. 5层台阶，8种
   斐波那契数列的思想。

3. 代码

func numWays(n int) int {
    if n == 1 || n == 0 {
        return 1
    }
    if n == 2 {
        return 2
    }
    a := 1
    b := 2
    for i := 2; i < n; i++ {
        b = a + b
        a = b - a
        b %= 1000000007
    }
    return b
}