这个题本身不难,但是涉及一个思想。
斐波那契:F(n)=F(n-1)+F(n-2),所以可以用递归,但是时间复杂度为O(2^N)
就是对一个时间复杂度为O(2^N)的递归算法应该如何优化:用个数组进行存储!
定义一个长度为N的数组 S。从小到大,将求得的数据存放到数组S[i]中。
由此,下一题:
题目描述
一只青蛙一次可以跳上1级台阶,也可以跳上2级。求该青蛙跳上一个n级的台阶总共有多少种跳法(先后次序不同算不同的结果)。
思路:
本来可以用递归,但是递归时间复杂度太高。改用一个数组存储 跳 n-1与n-2 级台阶的跳法
# -*- coding:utf-8 -*-
class Solution:
def jumpFloor(self, number):
# write code here
ans = [1]*(number+1)
for i in range(2,number+1):
ans[i] = ans[i-1]+ans[i-2]
return ans[number]
此时时间和空间复杂度都为O(N) , 可以看出这个数组每次只用到了index i 的前两个数。再改进一下:
class Solution:
def jumpFloor(self, number):
# write code here
ans0 = 1
ans1 = 1
for i in range(2,number+1):
temp = ans0 + ans1
ans0 = ans1
ans1 = temp
return ans1
这时 时间复杂度O(N), 空间复杂度 O(1)
问题变形:
一只青蛙一次可以跳上1级台阶,也可以跳上2级……它也可以跳上n级。求该青蛙跳上一个n级的台阶总共有多少种跳法。
思路:
用一个数组F存放 n级台阶的跳法
则 F[n] = F[0] + F[1] + ... + F[n-2]+ F[n-1]
由于 F[n-1] = F[0] + F[1] + ...+ F[n-2]
所以 F[n] = 2*F[n-1]
class Solution:
def jumpFloorII(self, number):
ans = 1
for i in range(2,number+1):
ans = 2*ans
return ans