剑指Offer（七）——贪心

1、跳台阶扩展问题

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

思路

方法一：暴力方法

设f[i] 表示 当前跳道第 i 个台阶的方法数。那么f[n]就是所求答案。

假设现在已经跳到了第 n 个台阶，那么前一步可以从哪些台阶到达呢？

如果上一步跳 1 步到达第 n 个台阶，说明上一步在第 n-1 个台阶。已知跳到第n-1个台阶的方法数为f[n-1]

如果上一步跳 2 步到达第 n 个台阶，说明上一步在第 n-2 个台阶。已知跳到第n-2个台阶的方法数为f[n-2]

。。。

如果上一步跳 n 步到达第 n 个台阶，说明上一步在第 0 个台阶。已知跳到 第0个台阶的方法数为f[0]

那么总的方法数就是所有可能的和。也就是f[n] = f[n-1] + f[n-2] + ... + f[0]

显然初始条件f[0] = f[1] = 1

所以我们就可以先求f[2]，然后f[3]...f[n-1]， 最后f[n]

解法

# -*- coding:utf-8 -*-
class Solution:
    def jumpFloorII(self, number):
        # write code here
        if number==0 or number==1:
            return 1
        f=[0 for i in range(number+1)]
        f[0]=f[1]=1
        for i in range(2,number+1):
            for j in range(0,i):
                f[i]+=f[j]
        return f[number]

上述解法时间复杂度 o(n2) ，空间复杂度 o(n)

优化

f[n] = f[n-1] + f[n-2] + ... + f[0]

f[n-1] = f[n-2] + f[n-3] + ... + f[0]

则f[n]=2*f[n-1],f[0]=f[1]=1

等比数列求和

# -*- coding:utf-8 -*-
class Solution:
    def jumpFloorII(self, number):
        # write code here
        n = 1
        for i in range(2, number+1):
            n = 2*n
        return n