一只青蛙一次可以跳上1级台阶,也可以跳上2级……它也可以跳上n级。求该青蛙跳上一个n级的台阶(n为正整数)总共有多少种跳法。
v1 – 偷懒的数学公式法
看到题目首先想到虽然青蛙n阶有几种跳法是未知的,但是肯定是从n-1,n-2,…0阶上来的,那么他跳跃的种类肯定是之前所有的阶数的跳法种类,写为kinds(n) = kinds(n-1) + kinds(n-2) + … + kinds(1) + 1, kinds(1)=1
这看起来就像是一个斐波那契数列的问题,如果递归肯定会有栈溢出和超时的问题。
如果尝试循环解决,从最小的kinds(2)开始遍历,那么对每个n都需要计算n个和,那么需计算(1+2+3+4+…+n)次,需要O(n(n-1)/2) =>O(n^2)的复杂度。感觉也不是很简洁。
计算前几个的结果有,kinds(1)=1,kinds(2)=2,kinds(3)=4, … kinds(n) = 2^(n-1),直接返回即可,
class Solution:
def jumpFloorII(self , number: int) -> int:
return 2**(number-1)
v2 – 动态规划法
写为kinds(n) = kinds(n-1) + kinds(n-2) + … + kinds(1) + 1, kinds(1)=1
简单按照直接的思路写个循环吧:
def jumpFloorII(self , number: int) -> int:
jumps = [0]*20
jumps[1] = 1
for n in range(2,number+1):
for k in range(1,n):
jumps[n] = jumps[n] + jumps[k]
# print(n, k, jumps[n])
jumps[n] = jumps[n] + 1
return jumps[number]