剑指offer - 第n个斐波那契数-青蛙跳格子

这个题本身不难，但是涉及一个思想。

斐波那契：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