剑指 Offer 10- II. 青蛙跳台阶问题 [简单]

剑指 Offer 10- II. 青蛙跳台阶问题 [简单]

题目描述

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

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

示例:

输入：n = 2
输出：2

输入：n = 7
输出：21

输入：n = 0
输出：1

限制：

0 <= n <= 100

解题思路：

此类求 多少种可能性 的题目一般都有 递推性质 ，即f*(n) 和 f(n-1)* …f(1) 之间是有联系的。

设跳上 n 级台阶有 f(n) 种跳法。在所有跳法中，青蛙的最后一步只有两种情况： 跳上 1 级或 2 级台阶。

• 当为 1 级台阶： 剩 n−1 个台阶，此情况共有 f(n−1) 种跳法；
• 当为 2 级台阶： 剩 n−2 个台阶，此情况共有 f(n−2)种跳法。

f(n) 为以上两种情况之和，即 f(n)=f(n−1)+f(n−2)，以上递推性质为斐波那契数列。本题可转化为 求斐波那契数列第 n 项的值 ，唯一的不同在于起始数字不同。

• 青蛙跳台阶问题：f(0)=1 ,f(1)=1 ,f(2)=2；
• 斐波那契数列问题： f(0)=0 ,f(1)=1 , f(2)=1 。

复杂度分析：

同斐波那契数列

代码：

//记忆化递归法
var result = map[int]int{}
func numWays(n int) int {
	if n == 0 {
		return 1
	}else if n==1 {
		return 1
	}else{
		// 递归法的优化 重复遇到某数字则直接取用，避免了重复的递归计算
		if _,ok:=result[n];!ok{
			result[n]=  numWays(n-1) + numWays(n-2)
		}
		return result[n]%1000000007
	}
}

// 动态规划
func numWaysV2(n int) int {
	dp := make(map[int]int,n)
	dp[0] = 1
	dp[1] = 1
	for i := 2; i <= n; i++{
		dp[i] = dp[i-1] + dp[i-2]
		dp[i] %= 1000000007
	}
	return dp[n]
}


// 动态规划 优化内存版本
func numWaysV3(n int) int {
	 dpA := 1
	 dpB := 1
	 var sum int
	for i := 0; i <  n; i++{
		sum = (dpA + dpB)%1000000007
		dpA = dpB
		dpB = sum
	}
	return dpA
}